[ 프로그래머스 ] 연속 부분 수열 합의 개수

문제

---

https://school.programmers.co.kr/learn/courses/30/lessons/131701

풀이

---

원형으로 배치된 수열의 부분 수열 개수를 구하는 문제로, 인접한 숫자들을 일정 길이로 묶어 더해주면 된다. 다만 원형으로 배치되어 있기 때문에 일정 길이에 도달하지 못했는데 배열 끝에 도달한다면 다시 첫 번째 순서로 돌아오게 하고, 중복을 제외하기 위해 set을 사용하여 배열 크기만큼 길이로 도달했을 때 set의 크기를 return한다.

#include <string>
#include <vector>
#include <set>

using namespace std;

int solution(vector<int> elements) 
{
    int answer = 0;
    set<int> s;

    for (int i = 0; i < elements.size(); ++i)
    {
        int sum = 0;
        for (int j = i; j < i + elements.size(); ++j)
        {
            sum += elements[j % elements.size()];
            s.insert(sum);
        }
    }

    answer = s.size();

    return answer;
}