HDU 4427 Math Magic (优化DP，ICPC2012长春)

题目描述：

点击打开链接

题意要用K个数组成和为N并且LCM为M的方案数，最直接的想法dp[i][j][k]表示i个数和为j并且LCM为k时方案数，由于要最后的LCM为M所以你用来组合的数都必须是M的约数，所以就先预处理出M的约数然后状态转移dp[i+1][[j+x][lcm[x,p]]=sum(dp[i][j][p])(x表示M的某个约数)。但是我这个题错了N种错法才A，首先MLE然后换滚动数组后RE，发现求和时可能出现大于N的情况导致数组越界然后改完TLE，然后把LCM打表后WA，发现自己某个小地方变量用混最后才A，真是艰难。。

AC代码：

#include<iostream>
#include<time.h>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
const double PI=acos(-1);
const int SIZE=50;
const long long INF=0x3f3f3f3f3f3f3f3f;
const int MAXM=1000;
const int MOD=1000000007;

int n,m,k;
int fm[MAXM+10];
int dp[2][MAXM+10][MAXM+10];
int LCM[MAXM+10][MAXM+10];

int gcd(int x,int y)
{
    return y==0? x:gcd(y,x%y);
}
int lcm(int x,int y)
{
    return x/gcd(x,y)*y;
}
void init()
{
    //memset(LCM,0,sizeof(LCM));
    for (int i=1;i<=1000;i++) {
        for (int j=1;j<=1000;j++)
            LCM[i][j]=lcm(i,j);
    }
}
int main()
{
    init();
    while(scanf("%d%d%d",&n,&m,&k)!=EOF)
    {
        memset(fm,0,sizeof(fm));
        int cnt=0;
        for (int i=1;i<=m;i++) {
            if (m%i==0) fm[cnt++]=i;
        }
       // sort(fm,fm+cnt);
        memset(dp,0,sizeof(dp));
        int v=0;
        for (int i=0;i<cnt;i++)
            dp[v][fm[i]][fm[i]]=1;
        for (int i=1;i<k;i++) {
            memset(dp[v^1],0,sizeof(dp[v^1]));
             for (int j=i;j<n;j++) {
                for (int p=0;p<cnt;p++) {
                    int lm=fm[p];
                    if (dp[v][j][lm]==0) continue;
                    for (int q=0;q<cnt;q++) {
                        if (j+fm[q]>n) break;
                        int mu=LCM[lm][fm[q]];
                        dp[v^1][j+fm[q]][mu]=(dp[v^1][j+fm[q]][mu]+dp[v][j][fm[p]])%MOD;
                    }
                }
             }
             v=v^1;
        }
        printf("%d\n",dp[v][n][m]%MOD);
    }
    return 0;
}