lightoj 1068 - Investigation 数位DP

给定a，b，k,问[a,b]之间有多少个k的倍数并且各个位数的和也是k的倍数的数字。

因为a,b是int范围所以k>90肯定不成立。

接下来按照数位DP考虑肯定要考虑两个余数...，但是k这个数字取值的最小公倍数过于大，瞬间懵逼..

看了别人的才知道每次都初始化去做...仔细想想复杂度也不高...

#include<bits/stdc++.h>
using namespace std;
#define inf 0x7fffffff
#define ll long long
int num[20],k;
int dp[20][100][100];
int dfs(int len,int y1,int y2,int f)
{
if(len==0) return y1==0&&y2==0;
if(!f&&dp[len][y1][y2]!=-1)
return dp[len][y1][y2];
int ans=0;
int l=f?num[len]:9;
for(int i=0;i<=l;i++)
{
int x1=(y1*10+i)%k;
int x2=(y2+i)%k;
ans+=dfs(len-1,x1,x2,f&&i==l);
}
if(!f)
dp[len][y1][y2]=ans;
return ans;
}
int solve(int x)
{
int i=1;
while(x)
{
num[i++]=x%10;
x=x/10;
}
return dfs(i-1,0,0,1);
}
int main()
{
int a,b;
int t;
scanf("%d",&t);
for(int cas=1;cas<=t;cas++)
{
scanf("%d %d %d",&a,&b,&k);
printf("Case %d: ",cas);
if(k>90)
printf("0\n");
else
{
memset(dp,-1,sizeof(dp));
printf("%d\n",solve(b)-solve(a-1));
}
}
return 0;
}