[프로그래머스Lv0] 다음에 올 숫자 - 파이썬

[프로그래머스Lv0] 다음에 올 숫자 - 파이썬

프로그래머스

 

<문제접근법>

- 등차수열인지 등비수열인지를 구분해야한다.

등차수열이면 두번째 인덱스에서 첫번째 인덱스를 뺏을 때와

세번째 인덱스와 두번째 인덱스를 뺏을 때의 값이 동일해야한다.

 

등비수열이면 두번째 인덱스에서 첫번째 인덱스를 나누었을 때 몫과

세번째 인덱스와 두번째 인덱스를 나누었을 때 값이 동일해야한다.

 

등차수열이라면 마지막의 값에서 공차만큼 더해주고 

등비수열이라면 마지막의 값에서 공비만큼 곱해준다.

def solution(common):
    answer = 0
    if common[1]-common[0] == common[2]-common[1] :
        answer = common[-1]+ (common[1]-common[0])
    elif common[1]//common[0] == common[2]//common[1] :
        answer = common[-1]*(common[1]//common[0])
    return answer