수들의 합 5 코딩테스트 연습풀이 [백준2018/파이썬/수학]

문제 출처: https://www.acmicpc.net/problem/2018

2018번: 수들의 합 5

 

문제

어떠한 자연수 N은, 몇 개의 연속된 자연수의 합으로 나타낼 수 있다. 당신은 어떤 자연수 N(1 ≤ N ≤ 10,000,000)에 대해서, 이 N을 몇 개의 연속된 자연수의 합으로 나타내는 가지수를 알고 싶어한다. 이때, 사용하는 자연수는 N이하여야 한다.

예를 들어, 15를 나타내는 방법은 15, 7+8, 4+5+6, 1+2+3+4+5의 4가지가 있다. 반면에 10을 나타내는 방법은 10, 1+2+3+4의 2가지가 있다.

N을 입력받아 가지수를 출력하는 프로그램을 작성하시오.

 

입력

첫 줄에 정수 N이 주어진다.

 

출력

입력된 자연수 N을 몇 개의 연속된 자연수의 합으로 나타내는 가지수를 출력하시오

 

정답

import sys
input = sys.stdin.readline
N = int(input())
start = 1
end = 1
sum = 1
count = 1 

if N == 1 or N == 2:
  print(1)
  sys.exit()

while end < N//2 + 2:
  if sum == N:
    count += 1
    end += 1
    sum += end
  elif sum < N:
    end += 1
    sum += end
  else:
    sum -= start
    start += 1
print(count)

변수들을 1로 세팅하여, 두개의 포인터를 사용해 한자리씩 전진하면서 더하는 방식으로 문제를 풀었습니다.