백준 알고리즘 1731번 추론(python)

수열이 등차인지 등비인지는 맨 처음 3개의 숫자들만 보면 알 수 있다. 

두 번째 숫자 - 첫 번째 숫자 = 세 번째 숫자 - 두 번째 숫자 -> 등차수열 

두 번째 숫자 / 첫 번째 숫자 = 세 번째 숫자 / 두 번째 숫자 -> 등비수열

이 점을 이용해서 등차라면 마지막 숫자에 li[1]-li[0] 만큼 더하고,

등비수열이라면 마지막 숫자에 li[1]//li[0] 만큼 곱해주면 된다.  

li = []
for _ in range(int(input())):
    li.append(int(input()))
    
ans = li[-1]
if li[2]-li[1] == li[1]-li[0]:
    ans += (li[1]-li[0])
elif li[2]//li[1] == li[1]//li[0]:
    ans *= (li[1]//li[0])

print(ans)