本题思路和DP状态不难求,比较难的是该怎么优化。
数位DP的优化可参考下面的博客:
https://blog.csdn.net/wust_zzwh/article/details/52100392
简单的说,一般数位DP的优化有两种方法:
①将memset放置在多组数据外面,即程序每运行一次只需memset一次,这种方法要求,题目所给的约束条件是每个数本身的属性,与输入无关。
②相减,这种方法是稍微改变一下状态,使得题目的约束条件转化为数本身的属性,从而调用方法①来对时间进行优化。
本题就是采用②。刚开始不难想出,dp的状态可以是dp[pos][sum],表示枚举到第pos位时,前面的数字的总权值为sum,满足条件的数就是最终sum>=F(A)即可。但是这么做的话每输入一个样例都要memset一次。于是,考虑到对于每个数的总权值本身也是固定的,属于它本身的属性,因此我们可以稍微改变一下dp状态:
dp[pos][sum]表示枚举到第pos位时,还需凑权值为sum的数字的个数。即一开始sum初始为F(A),每枚举一位,就用sum减去相应权值就可以了,不难想出,这么处理之后,dp状态就与F(A)无关了,因此现在就可以采用①方法了。
代码如下:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
const int inf=0x3f3f3f3f;
ll dp[11][4601];
int digit[11],fa;
ll dfs(int pos,ll sum,ll base,bool limit)
{
if(pos==-1) return 1;
if(!limit&&dp[pos][sum]!=-1) return dp[pos][sum];
ll ret=0;
int up= limit? digit[pos]:9;
for(int i=0;i<=up;++i)
{
if(sum-base*i<0) continue;
ret+=dfs(pos-1,sum-base*i,base/2,limit&&i==digit[pos]);
}
if(!limit) dp[pos][sum]=ret;
return ret;
}
ll solve(ll n)
{
int pos=0;
while(n)
{
digit[pos++]=n%10;
n/=10;
}
ll base;
for(int i=0;i<pos;++i)
if(!i) base=1;
else base*=2;
return dfs(--pos,fa,base,1);
}
int main()
{
int T,kase=0;
scanf("%d",&T);
memset(dp,-1,sizeof(dp));
while(T--)
{
ll a,b;
scanf("%lld%lld",&a,&b);
ll temp=1; fa=0;
while(a)
{
fa+=temp*(a%10);
a/=10; temp*=2;
}
printf("Case #%d: %lld\n",++kase,solve(b));
}
return 0;
}
小结:当数位DP时间卡得很严时,可以设法将题目要求的约束条件转化为数本身的属性,从而对时间进行优化。