zoj 3662 Math Magic(DP)

挺水的dp，脑抽把mod多打了个0结果弹了2次。

好吧，我们记录dp[i][j][k]为选i个数字，和为j,lcm为k的个数有多少种

接下来枚举第i+1个数字进行转移就行了，开三维会MLE，可以用滚动数组做。

我的做法有点类似背包，将lcm和sum从大到小枚举，这样避免重复。

#include <iostream>
#include <cstdio>
#include <cstring>
#define mod 1000000007
using namespace std;
int dp[1010][40];
int f[1010][1010];
int n,m,k;
int a[1010],aa[1010];
int gcd(int a,int b){
    if(f[a][b]!=-1) return f[a][b];
    if(b) return f[a][b]=gcd(b,a%b);
    else return f[a][b]=a;
}
int lcm(int a,int b){
    return a/gcd(a,b)*b;
}

int main(){
    memset(f,-1,sizeof f);
    while(~scanf("%d%d%d",&n,&m,&k)){
        memset(dp,0,sizeof dp);
        int ct=0;
        for(int i=1;i<=m;i++){
           if(m%i==0) a[ct++]=i;
        }
        for(int i=0;i<ct;i++) aa[a[i]]=i;
        int cur=0;
        for(int i=0;i<ct;i++) dp[a[i]][i]=1;
        for(int i=2;i<=k;i++,cur=1-cur){
            for(int j=ct-1;j>=0;j--){//i-1 lcm
               for(int p=n;p>=0;p--){//i-1 sum
                  if(!dp[p][j]) continue;
                  for(int q=0;q<ct&&a[q]+p<=n;q++){
                      int x=lcm(a[j],a[q]);
                      //cout<<p<<" "<<a[j]<<" "<<a[q]<<endl;
                      dp[p+a[q]][aa[x]]=(dp[p+a[q]][aa[x]]+dp[p][j])%mod;
                  }
                  dp[p][j]=0;
               }
            }
        }
        printf("%d
",dp[n][ct-1]);
    }
    return 0;
}