[백준] #1731 추론 python

2023. 2. 8. 06:00

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

 

1731번: 추론

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

www.acmicpc.net

 

📕 설명 📕

공차인 경우와 공비인 경우를 판별하여 풀이한다.

 

🧑🏻‍💻 나의 풀이 🧑🏻‍💻

N = int(input())
N_list = []
tmp = []
div = []
for i in range(N):
    N_list.append(int(input()))
    if i > 0:
        tmp.append(N_list[i] - N_list[i-1])
        div.append(N_list[i] // N_list[i-1])
if len(list(set(tmp))) == 1:
    print(N_list[N-1] + tmp[0])
else:
    print(N_list[N-1] * div[0])

BELATED ARTICLES

more