문제풀이/BOJ

[Python] BOJ/백준 1731번 추론

서채리 2021. 6. 16. 18:20

[문제]

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

 

1731번: 추론

등차가 정수인 등차수열 (어떤 수에 차례대로 일정한 수를 더해서 이루어지는 수열) 은 2개의 숫자로 나타낼 수 있다. P는 수열의 첫 번째 수이고, Q는 그 다음수가 되기 위해 바로 전의 수에 더해

www.acmicpc.net

 


[풀이]

수열의 원소들을 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]))