[Python] BOJ/백준 1731번 추론

[문제]

https://www.acmicpc.net/problem/1731

1731번: 추론

 

[풀이]

수열의 원소들을 num 리스트에 담는다. 원소들은 항상 3 이상이기 때문에 num[2]까지 이전 인덱스 원소와 비교해 등차수열인지 등비수열인지 찾는다. 

 

[코드]

if __name__ == "__main__":
    num = []
    for _ in range(int(input())):
        num.append(int(input()))

    if num[1] - num[0] == num[2] - num[1]:
        print(num[len(num) - 1] + (num[1] - num[0]))
    elif num[1] // num[0] == num[2] // num[1]:
        print(num[len(num) - 1] * (num[1] // num[0]))