数位dp例题——找数

找数

题面

给你两个数L,R,请求出在区间[L,R]中,有多少个数符合十进制下任意相邻K位数字的值两两互不相同,数不足K位则取全部数位。

一看这道题我们就知道是数位dp,但有k位,果断跳过,
赛后:原来k这么小,2<=k<=5,…
警示:一定要看数据范围


显然我们对于k的情况暴力枚举即可

        if(K==2){
			if(i==d) continue;
			 res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}
		if(K==3){
			if(i==d||i==c) continue;
			res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}
		if(K==4){
			if(i==b||i==c||i==d) continue;
			res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}
		if(K==5){
			if(i==a||i==b||i==c||i==d) continue;
			res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}

然后愉快的dp

#include
using namespace std;

typedef long long ll;
const int POS=20;
ll dp[POS][11][11][11][11];
ll l,r,K;
int num[POS];

ll dfs(int dep,bool limit,int a,int b,int c,int d){
	if(dep==0) return 1;
	if(!limit&&dp[dep][a][b][c][d]!=-1) return dp[dep][a][b][c][d];
	int up=limit?num[dep]:9;
	ll res=0;
	for(int i=0;i<=up;i++){
	    if(d==10&&i==0) res+=dfs(dep-1,limit&&i==up,10,10,10,10);
		else
		{
		if(K==2){
			if(i==d) continue;
			 res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}
		if(K==3){
			if(i==d||i==c) continue;
			res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}
		if(K==4){
			if(i==b||i==c||i==d) continue;
			res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}
		if(K==5){
			if(i==a||i==b||i==c||i==d) continue;
			res+=dfs(dep-1,limit&&i==up,b,c,d,i);
		}
	   }
	}
	if(!limit){
		dp[dep][a][b][c][d]=res;	
	}
	return res;
}
ll solve(ll x){
	int cnt=0;
	while(x){
		num[++cnt]=x%10;
		x/=10;
	}
	return dfs(cnt,true,10,10,10,10);
}
int main(){
	scanf("%lld%lld%d",&l,&r,&K);	memset(dp,-1,sizeof(dp));
	printf("%lld",(solve(r)-solve(l-1)));
}

P.S 一定要注意判断前导0,调了很久

你可能感兴趣的:(例题,动态规划)