CSU 1425 NUDT校赛 I题 Prime Summation

#include <iostream>
#include <cstdio>
#include <cstring>
#define N 210
#define ll unsigned long long
using namespace std;
ll n,k;
int dp[N][N],count[N],vis[N];
ll tot[N];
int tmp[N],prime[N],cnt;
void init()
{
    for (int i=0;i<N;i++)
        tmp[i]=1;
    for (int i=2;i<N;i++){
        for (int j=2;j*i<N;j++){
            tmp[i*j]=0;
        }
    }
    cnt=0;
    for (int i=2;i<N;i++){
        if (tmp[i]) prime[cnt++]=i;
    }
//    for (int i=0;i<cnt;i++)
//        cout<<prime[i]<<endl;
    memset(tot,0,sizeof tot);
    memset(count,0,sizeof count);
}
void solve()
{
    dp[2][count[2]++]=2;
    dp[3][count[3]++]=3;
    tot[2]=tot[3]=1;
    tot[0]=1;
    for (int i=4;i<N;i++){
        memset(vis,0,sizeof vis);
        if (tmp[i]){
            dp[i][count[i]++]=i;
            vis[i]=1;
            tot[i]++;
        }
        int up=lower_bound(prime,prime+cnt,i-2)-prime;
        while (prime[up]>i-2) up--;
        for (int j=up;j>=0;j--){
            int num=prime[j];
            bool flag=0;
            for (int k=count[i-num]-1;k>=0;k--){
                if (dp[i-num][k]>num) break;
                flag=1;
                tot[i]+=tot[dp[i-num][k]];
            }
            if (flag)
                dp[i][count[i]++]=num;
        }
    }
    tot[2]=tot[3]=1;
    for(int i=4;i<N;i++){
        tot[i]=0;
        for (int j=0;j<count[i];j++){
            tot[i]+=tot[dp[i][j]];
        }
    }
    int test=10;
    for (int i=0;i<count[test];i++)
        cout<<i<<" !!! "<<dp[test][i]<<endl;
   for (int i=2;i<N;i++)
        cout<<i<<" "<<tot[i]<<endl;;
}
void print(ll num,ll ret)
{
    if (num==0) return;
    ll sum=0;
    int i;
    ll pre=0;
    cout<<num<<" "<<ret<<endl;
    for (i=0;i<count[num];i++){
        sum+=tot[num-dp[num][i]];
        if (sum>=ret) break;
        pre+=tot[num-dp[num][i]];
    }
    //if (sum>ret) i--;
   // printf("%d",dp[num][i]);
    //if (num-dp[num][i]>0) printf("+");
    ll nt=ret-pre;
    print(num-dp[num][i],nt);

}
int main()
{
    init();
    solve();
    while (scanf("%llu%llu",&n,&k)!=EOF){
            printf("%llu\n",tot[n]);
            if (k>tot[n]) k=tot[n];
            printf("%d=",n);
            print(n,k);
            printf("\n");
    }
}
//200 15252874192862840692

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 210
using namespace std;
int f[N][N];
int prime[N],tot[N];
void init()
{
    memset(prime,0,sizeof prime);
    for (int i=2;i<N;i++)
        for (int j=i+i;j<N;j+=i) prime[j]=1; //预处理出素数
    memset(f,0,sizeof f);
    f[0][0]=1;
    for (int i=2;i<N;i++)//这里处理的非常简练，因为不存在的情况全部置为了0，所以省去了各种判断和限制，而且k是不会大于j的，使得计算结果不会出现重复。
        for (int j=2;j<=i;j++) if (!prime[j])
          for (int k=0;k<=j;k++){
            f[i][j]+=f[i-j][k];
          }
    for (int i=2;i<N;i++)
        for (int j=2;j<=i;j++)
         tot[i]+=f[i][j];
}
int n,k;
int main()
{
    init();
    while (scanf("%d%d",&n,&k)!=EOF){
        printf("%d\n",tot[n]);
        printf("%d=",n);
        if (k>tot[n]) k=tot[n];
        int cur=n,flag=0;;
        while (n>0){ //递推出式子，这里也处理的比较巧妙。值得学习
            if (k>f[n][cur]){
                k-=f[n][cur];
                cur--;
            }
            else
            {
                if (flag++) printf("+");
                printf("%d",cur);
                n-=cur;
            }
        }
        printf("\n");
    }
    return 0;
}