HDU 4427 Math Magic(DP)

题目链接

如果在现场赛，基本做不出。。。

这个题目感觉还是很有质量的，本来就知道这个题是DP，所以看题的时候就主动往状态转移上靠，然后看完之后觉得这个题很水啊。。。前i个的和为j，然后通过每一位上的因子递推就行了，然后就SB的敲了一个代码,WA了，想了一下，额，这种想法是错的。。。这样不能保证LCM = m,怎么才能保证LCM呢。。。想啊想，搞了一个更2的方法，状态还是刚那个样子，然后用组合的方式去枚举两个数，保证LCM = m,然后拍了一会，发现也不对，有重复的啊，难道要容斥。。。然后终于想到可以再加一维，带着当前状态的LCM转移，前i个和为j，当前的lcm为k，这样就可以了。。。4个for的复杂度+各种二货错误+卡内存 = 各种乱改+1MLE+2WA+2RE+2TLE，滚动数组滚的我纠结啊，终于1968ms卡过了。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <map>
using namespace std;
#define LL __int64
#define MOD 1000000007
LL dp1[1001][101],dp2[1001][101];
int o[1001],fact[101];
int ll[1001][1001];
int gcd(int a,int b)
{
    return b == 0?a:gcd(b,a%b);
}
int lcm(int a,int b)
{
    if(ll[a][b] > 0)//记忆化一下
    return ll[a][b];
    else
    {
        ll[a][b] = a*b/gcd(a,b);
        return ll[a][b];
    }
}
int main()
{
    int i,j,u,q,n,m,k,num,temp;
    LL ans;
    while(scanf("%d%d%d",&n,&m,&k)!=EOF)
    {
        num = 1;
        memset(o,0,sizeof(o));
        if(k == 1)//这个特判 居然写错了。。。
        {
            if(n == m)
            printf("%d\n",1);
            continue;
        }
        if(k == 2)
        {
            ans = 0;
            for(i = 1; i <= n-1; i ++)
            {
                if(lcm(i,n-i) == m)
                    ans ++;
            }
            printf("%I64d\n",ans);
            continue;
        }
        for(i = 1; i <= m; i ++)
        {
            if(m%i == 0)
            {
                fact[num] = i;
                o[i] = num;
                num ++;
            }
        }
        for(i = 0; i <= n; i ++)
        {
            for(j = 0; j <= num-1; j ++)
            {
                dp1[i][j] = 0;
                dp2[i][j] = 0;
            }
        }
        for(i = 1; i <= num-1; i ++)
        {
            dp1[fact[i]][i] = 1;
        }
        for(i = 1; i <= k-1; i ++)
        {
            for(j = 1; j <= n; j ++)
            {
                for(q = 1; q <= num-1; q ++)
                {
                    for(u = 1; u <= num-1; u ++)
                    {
                        if(dp1[j][q])
                        {
                            temp = lcm(fact[q],fact[u]);
                            if(j + fact[u] <= n&&temp <= m&&o[temp])
                            {
                                dp2[j+fact[u]][o[temp]] = (dp2[j+fact[u]][o[temp]]+dp1[j][q])%MOD;
                            }
                        }
                    }
                }
            }
            for(j = 1; j <= n; j ++)
            {
                for(q = 1; q <= num-1; q ++)
                {
                    dp1[j][q] = dp2[j][q];
                    dp2[j][q] = 0;
                }
            }
        }
        printf("%I64d\n",dp1[n][num-1]);
    }
    return 0;
}