【数位DP】SWUST 2014校赛 J:2014

总时间限制:1000ms   内存限制:32768kB

描述

今年是2014年，所以小明喜欢2014的每一位数字（即：2,0,1,4）,小明想知道在区间[l，r](包括l和r)中有多少个数中含有这4个数字（数字无前缀零）。

输入

多组数据。
每组数据输入2个数l，r（0

输出

输出占一行，即区间[l，r]（包括l和r）中包含的满足条件的数的个数

样例输入

1 10
100 1024

样例输出

0
1

套用模板，注意前导零的处理

#include
#include
#include
using namespace std;


int dp[20][2][2][2][2];//从左到右依次为len，是否含2，是否含0，是否含1，是否含4
int num[20];
int l,r;

int dfs(int pos,int s2,int s0,int s1,int s4,bool limit,bool fzero){
	if(pos==-1)return s2&&s0&&s1&&s4;
	if(!limit&&!fzero&&~dp[pos][s2][s0][s1][s4])return dp[pos][s2][s0][s1][s4];
	int end =limit?num[pos]:9;
	int ans=0;
	for(int i=0;i<=end;++i){
		int now2=s2,now0=s0,now1=s1,now4=s4;
		if(s2==0){
			if(i==2)now2=1;
		}
		if(s0==0){
			if(!fzero&&i==0)now0=1;
		}
		if(s1==0){
			if(i==1)now1=1;
		}
		if(s4==0){
			if(i==4)now4=1;
		}
		ans+=dfs(pos-1,now2,now0,now1,now4,limit&&i==end,fzero&&!i);
	}
	return (limit || fzero) ? ans : dp[pos][s2][s0][s1][s4] = ans;
}

int cal(int n){
	int len=0;
	while(n){
		num[len++]=n%10;
		n/=10;
	}
	return dfs(len-1,0,0,0,0,1,1);
}


int main(){
	
	memset(dp,-1,sizeof(dp));

	while(~scanf("%d%d",&l,&r)){
		printf("%d\n",cal(r)-cal(l-1));
	}
	return 0;
}