1. 程式人生 > >hdu 4427 Math Magic dp

hdu 4427 Math Magic dp

長春現場賽的題目,dp[i][j][k]表示公倍數為i,和為j,用了k個數的方法數。

但是狀態太多,注意到只有m的約數的公倍數才有可能是m,那麼可以預處理出m的所以約數,然後只遍歷這些狀態,再稍微優化一下常數就可以過了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=1e3+9,N=1000,mod=1e9+7;
int n,m,K;
int stack[maxn],end;
int dp[2][maxn][maxn],c[maxn][maxn];
int mm[maxn];
int gcd(int a,int b)
{
    int tmp;
    while(a%b)
    {
        tmp=a%b;
        a=b;
        b=tmp;
    }
    return b;
}
void get()
{
    for(int i=1;i<=N;i++)
    for(int j=1;j<=N;j++)
    c[i][j]=i*j/gcd(i,j);
}
int main()
{
//    freopen("in.txt","r",stdin);
    get();
    while(scanf("%d%d%d",&n,&m,&K)!=EOF)
    {
        end=0;
        for(int i=1;i<=m;i++)
        if(m%i==0)
        stack[++end]=i;
        memset(dp,0,sizeof(dp));
        int now=1,to=0;
        dp[to][1][0]=1;
        for(int k=0;k<K;k++)
        {
            now=now^1,to=to^1;
            for(int kk=1,i=stack[1];kk<=end;kk++,i=stack[kk])
            memset(dp[to][i],0,sizeof(dp[to][i]));
            for(int kk=1,i=stack[1];kk<=end;kk++,i=stack[kk])
            for(int j=0;j<=n;j++)
            if(dp[now][i][j])
            for(int p=1;p<=end&&stack[p]+j<=n;p++)
            if(c[i][stack[p]]<=m)
            {
//                cout<<c[i][stack[p]]<<endl;
                dp[to][c[i][stack[p]]][j+stack[p]]+=dp[now][i][j];
                if(dp[to][c[i][stack[p]]][j+stack[p]]>mod)
                dp[to][c[i][stack[p]]][j+stack[p]]-=mod;
            }
        }
//        cout<<tt<<endl;
        cout<<dp[to][m][n]<<endl;
    }
    return 0;
}