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;

}

posted @ 2016-08-07 11:53  幻世沉溺  阅读(204)  评论(0编辑  收藏  举报