HDU 5787 K-wolf Number
题意:l-r之间有多少个数,其连续k位不存在相同的数字
分析:数位dp,从高位开始向低位进行枚举。因为连续k个数字不相同,在枚举一个数字的时候,
要知道前k-1位的内容,这可以用一个4维的数组表示,再一点,前缀0的存在,
要区分现在枚举该位的0是不是前缀0,用一个标志记录bo就行,还有一点,就是前面
枚举的是不是比上限要小,比如统计255,如果第一个枚举1,那么剩下两位怎么放
都行,如果是2,那么加一个控制f,判断是否是否前面比上限小
*/
输入:
1 1 2 20 100 5
输出:
1 72
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll L,R,k;
int d[20];
ll dp[20][12][12][12][12];
ll dfs(int dep,int f,int now[],int bo) {
int x = now[0], y = now[1], z = now[2], e = now[3];
if(dep < 0) return 1;
if(f&&dp[dep][x][y][z][e]!=-1) return dp[dep][x][y][z][e];
if(f){
ll& ret = dp[dep][x][y][z][e];
ret=0;
for(int i = 0; i <= 9; ++i) {
int OK = 1;
for(int j = 0; j < k-1; ++j) if(now[j] == i) {OK = 0; break;}
if(OK == 0) continue;
if(!bo && !i) ret += dfs(dep-1,1,now,0);
else {
int tmpnow[6];
for(int j = 0; j <= 4; ++j) tmpnow[j] = now[j+1];
tmpnow[5] = 10;
tmpnow[k-2] = i;
ret += dfs(dep-1,1,tmpnow,1);
}
}
return ret;
}else {
ll ret = 0;
for(int i = 0; i <= d[dep]; ++i) {
int OK = 1;
for(int j = 0; j < k-1; ++j) if(now[j] == i) {OK = 0; break;}
if(OK == 0) continue;
if(!bo && !i) ret += dfs(dep-1,i<d[dep],now,0);
else {
int tmpnow[6];
for(int j = 0; j <= 4; ++j) tmpnow[j] = now[j+1];
tmpnow[5] = 10;
tmpnow[k-2] = i;
ret += dfs(dep-1,i<d[dep],tmpnow,1);
}
}
return ret;
}
}
ll solve(ll x) {
memset(dp,-1,sizeof(dp));
int len = 0;
while(x) {
d[len++] = x % 10;
x /= 10;
}
int now[6];
for(int i = 0; i <= 5; ++i) now[i] = 10;
return dfs(len-1,0,now,0);
}
int main(){
while(cin>>L>>R>>k)
cout<<solve(R)-solve(L-1)<<endl;
return 0;
}