数位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，调了很久