HDU4734,F(x)(数位DP+优化)

本题思路和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时间卡得很严时,可以设法将题目要求的约束条件转化为数本身的属性,从而对时间进行优化。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值