[PS] 백준#1024 수열의 합 / 파이썬

알고리즘 문제풀이

• 문제 : 수열의 합
• 해결 : solved
• 분류 : 수학(수열)

풀이방법

1. 등차수열 공식을 활용하여 식을 만든다. a + (n-1)d
2. L부터 100까지 순회하면서 수열의 합이 특정값을 만족하는지 검사한다.
3. 결과를 출력한다.

소스코드

N, L = map(int, input().split())
x = []

def get_arr(length, N):
    x = N - (length * (length+1) // 2)
    return x / length

for i in range(L, 101):

    a = get_arr(i, N)
    if int(a) == a :
        a = int(a)
        sub_sum = sum(range(a+i+1)) - sum(range(a+1))
        if sub_sum == N :
            x = (list(range(a + 1, a + i + 1)))
            x = list(map(str, x))
            break

print(' '.join(x) if x != [] and len(x) <= 100 else -1)