HDU - 2089 - 不要62 ( 数位dp )

题目链接：点击进入

思路

思路

pos //第pos位
pre//上一位值上的数
state//前一位是否为6
lead//是否前导零
limit//是否是最高位
要求没有62以及4，没有4的话，在枚举的时候判断一下，不枚举4就可以保证状态合法，而对于62的话，涉及到两位，当前一位是6或者不是6这两种情况计数是不相同的，要用状态来记录不同的方案数。

代码

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define max(a,b) a>b?a:b
//#define min(a,b) a<b?a:b
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=2e4+10;
const int mod=998244353;
const int Mod=1e9+10;
const double eps=1e-9;
const int N=1e6+10;
int a[10];
int dp[10][2];
int dfs(int pos,int pre,int state,bool lead,bool limit)
{
	if(pos==-1) return 1;
	if(!limit&&!lead&&dp[pos][state]!=-1) return dp[pos][state];
	int up=limit?a[pos]:9;
	int res=0;
	for(int i=0;i<=up;i++)
	{
		if(i==2&&pre==6) continue;
		if(i==4) continue;
		res+=dfs(pos-1,i,i==6,lead&&i==0,limit&&i==a[pos]);
	} 
	if(!limit&&!lead) dp[pos][state]=res;
	return res;
}
int solve(int x)
{
	int pos=0;
	while(x)
	{
		a[pos++]=x%10;
		x/=10;
	}
	return dfs(pos-1,-1,0,1,1);
}
int main()
{	
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	int l,r;
	memset(dp,-1,sizeof(dp));
	while(~scanf("%d%d",&l,&r))
	{
		if(l==0&&r==0) break;
		if(l) 
			printf("%d\n",solve(r)-solve(l-1));
		else
			printf("%d\n",solve(r),solve(l)); 
	}
    return 0;
}