(Python/LV0) 연속된 수의 합

문제 출처:https://school.programmers.co.kr/learn/courses/30/lessons/120923

프로그래머스

풀이:

from collections import deque
def solution(num, total):
    deque_num = deque([i for i in range(num)])
    deque_num_sum = sum(deque_num)
    while deque_num_sum < total:
        x = deque_num.popleft()
        y = deque_num[-1] +1
        deque_num.append(y)
        deque_num_sum -= x
        deque_num_sum += y

    while deque_num_sum != total:
        x = deque_num.pop()
        deque_num_sum -= x
        y = deque_num[0]-1
        deque_num.appendleft(y)
        deque_num_sum += y

    return list(deque_num)

#90 / 100
# 다시 생각해봐야겠다

 

 

테스트 케이스 1번이 틀린다 왜 틀리는거지,,

 

num == 1일 경우 생각해야함

from collections import deque
def solution(num, total):
    if num ==1:
        return [total]
    deque_num = deque([i for i in range(num)])
    deque_num_sum = sum(deque_num)
    while deque_num_sum < total:
        x = deque_num.popleft()
        y = deque_num[-1] +1
        deque_num.append(y)
        deque_num_sum = deque_num_sum - x + y

    while deque_num_sum > total:
        x = deque_num.pop()
        y = deque_num[0]-1
        deque_num.appendleft(y)
        deque_num_sum = deque_num_sum - x + y

    return list(deque_num)