连续质数和 BILIBILI 20208.13 算法笔试 c++ 思路

题目 ： 有一些正整数可以表示成若干连续质数的和（可以仅有一个）。例如，53可以表示成5+7+11+13+17和53自身，而41可以表示成2+3+5+7+11+13, 11+13+17和41自身；因此53和41表示成连续质数和的方案数分别是2和3。但是20就没有表示成连续质数的和。

输入一系列的nn，编程求它们的连续质数和的方案数。

思路：记录小于当前值的质数，同时记录前缀和，加入map，计算方案值，一趟扫描。

#include <vector>
#include <unordered_map>
#include <iostream>
using namespace std;

int main()
{
    int n;
    while(cin>>n){
        if (n==2) {
            cout<<1<<endl;
            return 0;
        }
        int i = 3;
        vector<int> z_list = {2};
        unordered_map<int,int> mp;
        mp[0] = 1;
        int sum = 2;
        int res = 0;
        while(i <= n) {
            bool flag = true;
            for(int j : z_list){
                if(i % j == 0) {
                    flag = false;
                    break;
                }
            }
            if(flag) {
                z_list.push_back(i);
                sum += i;
                mp[sum]++;
                if (mp.find(sum - n) != mp.end()) {
                    res += mp[sum - n];
                }
            }
            i++;
        }
        cout<<res;
    }
    return 0;
}

如有问题请指正