[Programmers] Lv2. 숫자의 표현 | C++

🔗문제 보러가기

프로그래머스

 

👨‍💻풀이 과정

 

1. for문을 통해 자연수 1부터 차례대로 합계에 더하고, Queue에도 원소를 추가합니다.
2. 만약 합계가 n과 같다면 answer + 1를 해주고, n < 합계가 되는 상황이라면 Queue에서 하나씩 pop 하여 합계에서 뺍니다.
3. Queue의 원소를 pop하여 뺀 합계가 n과 같다면 answer + 1을 해 주고, 아니라면 큐가 비거나, 합계 < n이 되는 상황까지 진행해줍니다.

 

✏️소스 코드 및 결과

#include <string>
#include <queue>
using namespace std;

int solution(int n) {
    queue<int> myQueue;
    int answer = 0;
    int sum = 0;

    for (int i = 1; i <= n; i++) {
        sum += i;
        myQueue.push(i);

        if (sum == n)
            answer++;
        else if (n < sum) {
            while (!myQueue.empty()) {
                int num = myQueue.front();
                myQueue.pop();
                sum -= num;

                if (sum == n)
                    answer++;
                else if (sum < n)
                    break;
            }
        }
    }

    return answer;
}