HDU 4734 F(x)（数位dp）

题意：寻找 [0,B] 中满足条件的数的数量。条件：这个数的weigh<=F(A)

题解：数位dp。

一开始存了当前状态的weigh和F(A)，但是这样MLE了...

后来看了题解，发现存的是F(A)-weigh orz

#include<cstdio>
#include<cstring>

int fa;
int dig[15],d[15][5000],ba[15];

int dp(int po,int sum,bool limit) {
	if(po<0) return 1;
	if(!limit&&d[po][sum]!=-1) return d[po][sum];
	int to=limit?dig[po]:9;
	int ans=0;
	for(int i=0;i<=to;++i) {
		int nxtsum=sum-ba[po]*i;
		if(nxtsum<0) continue;
		ans=ans+dp(po-1,nxtsum,limit&&i==to);
	}
	if(!limit) d[po][sum]=ans;
	return ans;
}

int solve(int x) {
	int po=0;
	while(x) {
		dig[po++]=x%10;
		x/=10;
	}
	return dp(po-1,fa,1);
}

int main() {
	///init
	memset(d,-1,sizeof(d));

	///get ba
	ba[0]=1;
	for(int i=1;i<15;++i) ba[i]=ba[i-1]*2;

	int T;scanf("%d",&T);
	int ca=0;
	while(T--) {
		printf("Case #%d: ",++ca);

		///read
		int a,b;scanf("%d%d",&a,&b);

		///get fa
		fa=0;
		int base=1;
		while(a) {
			fa=fa+base*(a%10);
			a/=10;
			base<<=1;
		}

		///solve
		printf("%d\n",solve(b));
	}
	return 0;
}