[HDOJ6156] Palindrome Function（数位dp, 枚举）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6156

题意：算十进制下数字在[L,R]内用[l,r]进制表述下的数字贡献。

贡献有两种：回文数，贡献是进制k；不是回文数，贡献是1。

由于进制只有36个，枚举进制分别做统计回文数的数位dp即可，贡献按要求。

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn = 1010;
const int maxm = 55;
const LL mod = 1e9+7;
int digit[maxn], revert[maxn];
LL L, R, l, r;
LL dp[maxm][maxm][maxn][2];

LL dfs(int k, int s, int l, bool ok, bool flag) {
    if(l < 0) {
        if(ok) return k;
        return 1;
    }
    if(!flag && ~dp[k][s][l][ok]) return dp[k][s][l][ok];
    int pos = flag ? digit[l] : k - 1;
    LL ret = 0;
    for(int i = 0; i <= pos; i++) {
        revert[l] = i;
        if(i == 0 && s == l) {
            ret += dfs(k, s-1, l-1, ok, flag&&(i==pos));
        }
        else if(ok && l < (s + 1) / 2) {
            ret += dfs(k, s, l-1, i==revert[s-l], flag&&(i==pos));
        }
        else {
            ret += dfs(k, s, l-1, ok, flag&&(i==pos));
        }
    }
    if(!flag) dp[k][s][l][ok] = ret;
    return ret;
}

LL f(LL n, LL k) {
    if(n == 0) return k;
    int pos = 0;
    while(n) {
        digit[pos++] = n % k;
        n /= k;
    }
    return dfs(k, pos-1, pos-1, 1, 1);
}


signed main() {
    // freopen("in", "r", stdin);
    int T, _ = 1;
    scanf("%d", &T);
    memset(dp, -1, sizeof(dp));
    while(T--) {
        scanf("%lld%lld%lld%lld",&L,&R,&l,&r);
        LL ret = 0;
        for(int i = l; i <= r; i++) {
            ret += f(R, i) - f(L-1, i);
        }
        printf("Case #%d: %lld\n", _++, ret);
    }
    return 0;
}