[Programmers] 다음에 올 숫자 (Swift)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120924

프로그래머스

풀이

• 등비수열인지 등차수열인지 판별하는 로직이 있어야 할 것이라고 생각함
• Enum을 사용해서 타입을 정해줄 수 있지 않을까?!
• 1번 데이터와 2번 데이터의 차이를 먼저 계산하고 2번 데이터와 3번 데이터의 차이가 같다면 등차, 다르다면 등비 일 것이다!

소스코드

	import Foundation
	enum Sequence {
	case arithmetic
	case geometric
	}
	func solution(_ common:[Int]) -> Int {
	let d = common[1] - common[0]
	let sequenceType: Sequence = d == common[2] - common[1] ? .arithmetic : .geometric
	switch sequenceType {
	case .arithmetic:
	return common.last! + d
	case .geometric:
	return common.last! * (common[1] / common[0])
	}
	}

view raw 다음에 올 숫자.swift hosted with ❤ by GitHub

후기

다른분들의 풀이를 봤는데 1줄로 짠 코드를 봤다.

풀이한 로직은 거의 비슷했다고 생각이 들었음!