【DP】 HDU 4427 Math Magic

点击打开链接

题意：求K 个数的数列  ，和为N ， LCM 为M 的个数

因为M 的 除数很少 最多10 个  先处理出来就可以O(K*N*10*10) 处理了

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
#pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long LL;
typedef unsigned long long ULL;
const int INF = 1<<29;
const LL mod = 1e9+7;
const int MAXN = 110100;//点数的最大值
const int MAXM = 520;//边数的最大值
const double pi = acos(-1.0);
int dp[2][1010][1010];
int gcd(int a,int b)
{
if(b==0) return a;
else return gcd(b,a%b);
}
int lcm(int a,int b)
{
return a/gcd(a,b)*b;
}
void Mod(int &a,int b)
{
a+=b;
if(a>=mod) a-=mod;
}
int Lcm[1010][1010];
vector<int> di;
int main()
{
int n,m,k;
for(int i=1;i<=1e3;i++)
for(int j=1;j<=1e3;j++)
Lcm[i][j]=lcm(i,j);
while(scanf("%d%d%d",&n,&m,&k)!=EOF)
{
di.clear();
for(int i=1;i<=m;i++)
if(m%i==0) di.push_back(i);
memset(dp,0,sizeof(dp));
for(int i=0;i<di.size();i++)//起始状态
dp[0][di[i]][di[i]]=1;
int now=0,last;
for(int i=1;i<k;i++)
{
last=now;
now^=1;
for(int j=i;j<=n;j++)
{
for(int w=0;w<di.size();w++)
{
if(dp[last][j][di[w]]==0)
continue;
for(int d=0;d<di.size()&&j+di[d]<=n;d++)
{
if(m%Lcm[di[w]][di[d]]==0)
Mod(dp[now][j+di[d]][Lcm[di[w]][di[d]]],dp[last][j][di[w]]);
}
dp[last][j][di[w]]=0;//清空
}
}
}
printf("%d\n",dp[now]
[m]);
}
return 0;
}
/*
*/