프로그래머스

프로그래머스 lvl 2. 숫자의 표현 [C++]

soobinida 2023. 7. 2. 14:45

Finn은 요즘 수학공부에 빠져 있습니다. 수학 공부를 하던 Finn은 자연수 n을 연속한 자연수들로 표현하는 방법이 여러개라는 사실을 알게 되었습니다. 예를들어 15는 다음과 같이 4가지로 표현 할 수 있습니다.

  • 1 + 2 + 3 + 4 + 5 = 15
  • 4 + 5 + 6 = 15
  • 7 + 8 = 15
  • 15 = 15

 

자연수 n이 매개변수로 주어질 때, 연속된 자연수들로 n을 표현하는 방법의 수를 return하는 solution을 완성해주세요.

 

접근 point

  • n보다 작거나 같은 연속된 자연수
  • 투 포인터 사용

 

n = 10 일 때, 포인터 s e 를 생성하여, index = 0 부터 시작한다.

s 와 e 자리에 있는 값을 더한 후, n 보다 작으면 e 포인터를 올려 새로운 e 자리에 있는 값을 더하고, n 보다 크거나 같으면 s 포인터를 올리고 기존 s 자리에 있는 값을 빼준다. 

 

위 단계를 e ≤ n 일 때까지 반복한다.

 

#include <bits/stdc++.h>
using namespace std;

int solution(int n) {
    int ret = 0, s = 1, e = 1, sum = s;
    while(e <= n) {
        if(sum < n) {
            e++;
            sum += e;
        } else {
            if(sum == n) ret++;
            sum -= s;
            s++;
        }
    }
    return ret;
}