UVa 1213 Sum of Different Primes Solution

    #include <iostream>
    #include <vector>
    #include <algorithm>
    #include <stack>
    using namespace std;
    vector<int> pr={2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,79,83,89,97,101,103,107,109,113,127,131,137,139,149,151,157,163,167,173,179,181,191,193,197,199,211,223,227,229,233,239,241,251,257,263,269,271,277,281,283,293,307,311,313,317,331,337,347,349,353,359,367,373,379,383,389,397,401,409,419,421,431,433,439,443,449,457,461,463,467,479,487,491,499,503,509,521,523,541,547,557,563,569,571,577,587,593,599,601,607,613,617,619,631,641,643,647,653,659,661,673,677,683,691,701,709,719,727,733,739,743,751,757,761,769,773,787,797,809,811,821,823,827,829,839,853,857,859,863,877,881,883,887,907,911,919,929,937,941,947,953,967,971,977,983,991,997,1009,1013,1019,1021,1031,1033,1039,1049,1051,1061,1063,1069,1087,1091,1093,1097,1103,1109,1117};
    int kn(int n,int k,int i,vector<vector<vector<int> > > &dp){
        if(n==0){if(k==0)return 1;
        else return 0;}
        if(k==0||i==pr.size()||pr[i]>n)return 0;
        if(dp[n][k][i]!=-1)return dp[n][k][i];
        return dp[n][k][i]=kn(n,k,i+1,dp)+kn(n-pr[i],k-1,i+1,dp);
    }
    // void rec(vector<vector<int> > &dp,stack<int> &s,int i,int t,int c,vector<int> &d){
    //     if(i==0)return;
    //     if(dp[i][t]==dp[i-1][t])return rec(dp,s,i-1,t,c,d);
    //     else {
    //         s.push(i);
    //         return rec(dp,s,i-1,t-3*c*d[i],c,d);
    //     }
    // }
    vector<vector<vector<int> > > dp(1121,vector<vector<int> > (15,vector<int> (187,-1)));
    int main(){
        int n,k;
        while(cin>>n>>k){
            if(n==0&&k==0)return 0;
            cout<<kn(n,k,0,dp)<<endl;
        }
        return 0;
    }

Comments