[BOJ/백준] 5101 - Sequences [python]

문제 링크

5101번: Sequences

문제 풀이

등차 수열의 첫번째 값, 차, 검사할 값이 주어지는데, 검사할 값과 첫번째 값을 빼준 뒤, 차를 나누었을 때 나머지가 0이라면 검사할 값은 등차수열에 포함이 된다.

따라서 첫번째 값, 차, 검사할 값을 각각 a, b, c라고한다면, (c - a) % b == 0이 되는 값을 찾아주면 된다.

단, 주의할 점이 있는데, 나누어 떨어지는 경우만 찾게 되면, 3, 2, 1같은 경우에 불가능한 입력이지만 정답으로 출력되므로 (c - a) // b >= 0의 조건을 추가해준다.

정답 코드

while True :
    a, b, c = map(int, input().split())
    if a == b == c == 0 : break
    if (c - a) % b == 0 and (c - a) // b >= 0 : print((c - a) // b + 1)
    else : print("X")