[0단계] 다음에 올 숫자

[프로그래머스 문제 사이트]

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

프로그래머스

 

[문제]

[코드]

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

 

[해석]

1. count 변수 선언 이유
: count 변수는 등차수열인지 등비수열인지 구별해주기 위함.

 

2. if문의 의미
: count 변수를 배열 안의 뒤에서 두 번째 수를 더했을 경우, 마지막 배열 안의 원소 값과 같으면 등차수열로 인식하기 위함. 만약, 같지 않다면 결과는 else로 빠짐.

 

3. 이렇게 코드를 해도 될까?
: 어차피 문제에서의 조건은 등차수열 아니면 등비수열이므로 관계 없을 것임. 따라서, else로 빠졌다면, 등비수열로, count를 새로 정의해주고 마지막 값과 곱하여 리턴하면 끝!