[Algorithm]Baekjoon 1644 :: 소수의 연속합

https://www.acmicpc.net/problem/1644

1644번: 소수의 연속합

 
using namespace std;
 
int n;
bool isPrimes[4000001] = { false };
vector<int> primes;
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    cin >> n;
 
    if (n == 1)
    {
        cout << 0;
        return 0;
    }
 
    for (int i = 2; i <= n; ++i)
    {
        if (isPrimes[i] == false)
        {
            primes.push_back(i);
 
            for (int j = i + i; j <= n; j+=i)
            {
                isPrimes[j] = true;
            }
        }
    }
 
    int left = 0;
    int right = 0;
    int sum = primes[0];
    int answer = 0;
 
    while (right < primes.size())
    {
        if (sum == n)
        {
            ++answer;
            ++right;
            if (right < primes.size())
            {
                sum += primes[right];
            }
        }
        else if (sum < n)
        {
            ++right;
            if (right < primes.size())
            {
                sum += primes[right];
            }
        }
        else
        {
            sum -= primes[left];
            ++left;
        }
    }
 
    cout << answer;
}
 
 
 
 

 

특정숫자 n이 주어졌을 때, 소수의 연속합으로 해당 숫자를 만들 수 있는 경우의 개수를 구하는 문제이다.

연속합을 구한다는 부분에서 투포인터가 떠올랐고, 여러개의 소수를 구해야한다는 점에서 에라토스테네스의 체가 떠올랐다.

딱 이 두개만 알고있으면 이 문제는 아주 쉽다. 에라토스테네스의 체로 n까지의 소수들을 전부 구해준 후, 해당소수들이 들어있는 벡터의 인덱스로 투포인터를 적용해서 연속합들을 만들어주고, 그게 타겟값 n일 경우 ++answer 해주면 된다.