소수의 연속합(백준 1644)

링크

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

코드

#pragma warning(disable:4996)
#include <iostream>
#include <vector>
#include <algorithm>
#include <climits>
#include <map>
#include <queue>
#include<cmath>
#include <string>
#include <cstring>
#include <set>
#include<stack>
#define ull unsigned long long
#define ll long long
#define pa pair<int, int>
#define dpa pair<double,int>
#define ppa pair<int, pa>
#define INF 98765432
using namespace std;

int n;

vector<ll> primes;
vector<ll> primes_sum;

bool isPrime(ll inp) {
    for (int i = 0; primes[i] * primes[i] <= inp; i++) {
        if (inp % primes[i] == 0)
            return false;
    }
    return true;
}

int main(void) {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cin >> n;

    primes_sum.push_back(0);

    for (ll i = 2; i <= 4000000; i++) {
        bool flag = true;
        for (int j = 0; j<primes.size()&&primes[j] * primes[j] <= i; j++) {
            if (i % primes[j] == 0) {
                flag = false;
                break;
            }

        }
        if (flag) {
            primes.push_back(i);
            primes_sum.push_back(primes_sum[primes_sum.size() - 1] + i);
        }
            
    }

    int left = 0, right = 0, counter = 0;

    while (left <= right && right < primes_sum.size()) {
        if (primes_sum[right] - primes_sum[left]>n) {
            left++;
        }
        else if (primes_sum[right] - primes_sum[left] < n) {
            right++;
        }
        else {
            right++;
            counter++;
        }

    }
    cout << counter << endl;
    return 0;
}

조심할 부분

• 코드 상에서, 400만개를 도는, 전체 input 범위 검사 과정이 제일 큰 시간 값을 가지게 됩니다. O(NM)의 시간 복잡도를 가지므로 TLE(400만 * 20만 = 8천억) 가능성이 크나, 실제로 바깥 루프인 N 중, 모든 케이스를 도는 경우는 거의 없으므로 시간 내에 들어오는 것으로 보임.
• 소수의 부분합을 저장하는 primes_sum의 경우, 위 문제에서 two pointer를 사용하는 용례 상 초기값으로 0을 집어넣어주는 방식이 필요함.