题面
给你两个数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<bits/stdc++.h>
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,调了很久