[2016ACM多校] HDU5787 数位DP

题意

求[L,R]中这样一种数的个数，它的每连续K位都满足两两不相同。2<=K<=5

思路

K很小，典型的数位动态规划，为了方便写成记忆话搜索的形式。从高位往低位决策，传入之前临近的K位是哪些数，这一位是否可以自由枚举0~9，是否来自前导零。为了修改最近的K位所占用的数，要用一个队列或者其他数据结构维护填数的先后顺序。如果这一位可以自由枚举那么之后就都可以自由枚举，而且仅受制于前K位用过几个数(前导零不算，这里是可记忆化的关键)，记忆化可以自由枚举的情况，就不会超时。但K发生改变记忆化就不行了，要初始化。复杂度O(10^KlogN)。

AC代码 C

#include <stdio.h>
#include <string.h>

#define MAXN 20

int K;

char ns[MAXN];
int nt;
//将数字分解到字符串
void decompose(long long x)
{
    nt = 0;
    while(x)
    {
        ns[nt++] = x % 10;
        x /= 10;
    }
}

long long dp[MAXN][1<<10][2];   //dp[i][j][k]到第i位,stat为j,begin为k的方案数
int path[MAXN];
int pt;
//记忆化搜索,bit当前决策位,stat前K位数字,fe是否可自由枚举,begin前导0结束
long long dfs(const int bit,  const int stat, const char fe, const char begin)
{
    if(bit < 0)
        return 1;
    if(fe && dp[bit][stat][begin] >= 0)
        return dp[bit][stat][begin];
    long long cnt = 0;
    int top = fe ? 9 : ns[bit], next_stat;
    for(int i=0; i<=top; i++)
    {
        if(stat & 1 << i)
            continue;
        if(begin || i)
            path[pt++] = i;
        next_stat = pt < K ? stat : stat ^ 1 << path[pt - K];
        cnt += dfs(bit-1, next_stat | (begin || i) << i, fe||i<top, begin||i);
        if(begin || i)
            pt--;
    }
    if(fe)
        dp[bit][stat][begin] = cnt;
    return cnt;
}

int main()
{
    long long l, r;
    while(scanf("%I64d%I64d%d", &l, &r, &K) > 0)
    {
        memset(dp, -1, sizeof dp);
        decompose(r);
        r = dfs(nt-1, 0, 0, 0);
        decompose(l - 1);
        l = dfs(nt-1, 0, 0, 0);
        printf("%I64d\n", r - l);
    }
    return 0;
}