题意:给定一个区间[n,m],问在其中的不包含数字4,且不包含连续的62的数字有多少个?
思路:典型的数位dp。dp[i,j]表示开头为j的i位数中满足要求的数量。求和的时候先解析出数字,然后从高位往低位求和。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define clr(s,t) memset(s,t,sizeof(s));
using namespace std;
#define N 20005
int dp[9][12],digit[9];
int n,m;
void init(){
clr(dp, 0);
dp[0][0] = 1;
for(int i = 1;i<=7;i++)
for(int j = 0;j<=9;j++)
for(int k = 0;k<=9;k++)
if(!(j==4 || ((j==6)&&(k==2))))
dp[i][j] += dp[i-1][k];
}
int solve(int x){//[0,x-1]符合条件的个数
int len = 0;
int res = 0;
while(x){
digit[++len] = x%10;
x /= 10;
}
digit[len+1] = -1;
for(int i = len;i>=1;i--){
for(int j = 0;j<digit[i];j++)
if(!(j==4 || ((j==2)&&(digit[i+1]==6))))
res += dp[i][j];
if(digit[i] == 4 || ((digit[i]==2)&&(digit[i+1]==6)))//一个剪枝,当前位已经不满足了,不必往后求和啦
break;
}
return res;
}
int main(){
init();
while(scanf("%d %d",&n,&m) && (n+m))
printf("%d\n",solve(m+1)-solve(n));
}