[BOJ][B2]추론 - 1731

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

문제 전문은 링크 참조

문제가공

1. 첫번째 입력값 = 수열이 입력될 횟수
2. 입력값을 통해
   등차 또는 등비수열 인지 확인하여 마지막 입력 값의 다음 값을 출력

코드작성

• 등차, 등비를 확인 하기 위해서는 세개의 수열 값만 있으면 된다.
• 아래 공식이 성립하지 않으면 등비수열

cnt = int(input())
f,s,c = 0,0,0 #first,second,third
gap = 1
for i in range(cnt):
    val = int(input())
    if f == 0 : f = val
    elif s == 0 : s = val
    elif c == 0 : c = val

if(abs(f-s) == abs(s-c)): # 등차수열
    gap = abs(f-s)
    print(f +(gap * cnt))
else: #등비수열
    gap = s//f
    print(f * (gap**cnt))