hdu 3555 Bomb

链接：http://acm.hdu.edu.cn/showproblem.php?pid=3555

数位dp

#include<stdio.h>
#include<string.h>
#define ll long long
int bit[100];
ll dp[22][3];
//dp[i][0],[10^(i-1),10^i)内有多少个包含49的
//dp[i][1],[10^(i-1),10^i)内有多少个不包含49但以9开头
//dp[i][2],[10^(i-1),10^i)内有多少个不包含49
void inidp()
{
	int i;
	memset(dp,0,sizeof(dp));
	dp[0][2]=1;
	for(i=1;i<=20;i++)
	{
		dp[i][0]=dp[i-1][0]*10+dp[i-1][1];
		dp[i][1]=dp[i-1][2];
		dp[i][2]=dp[i-1][2]*10-dp[i-1][1];
	}
}
ll solve(ll n)
{
	int i,bit[40],len=0;
	while(n)
	{
		bit[++len]=n%10;
		n/=10;
	}
	bit[len+1]=0;
	ll ans=0;
	int flag=0;
	for(i=len;i>0;i--)
	{
		ans+=dp[i-1][0]*bit[i];
		if(flag)
			ans+=dp[i-1][2]*bit[i];
		else
			if(bit[i]>4) ans+=dp[i-1][1];
		if(bit[i+1]==4&&bit[i]==9) flag=1;
	}
	return ans;
}
int main()
{
	inidp();
	int cas;
	ll n;
	scanf("%d",&cas);
	while(cas--)
	{
		int i=0;
		scanf("%I64d",&n);
		printf("%I64d\n",solve(n+1)+i);//记住+1啊，，，WA了好多次才发现
	}
	return 0;
}