hdu 6156 Palindrome Function【数位dp求回文数】

题目

http://acm.hdu.edu.cn/showproblem.php?pid=6156

题意

求2～k进制下在【L，R】范围内的回文数

分析

数位dp求回文数，注意T范围很大，那么直接记忆进制即可。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

int a[70];
LL dp[70][70][40];

LL dfs(int len,int l,int r,bool limit,bool ok,int x)
{
    if(l<r) return !limit||(limit&&ok);
    if(!limit&&~dp[len][l][x])
        return dp[len][l][x];
    LL ret=0;
    int mx=limit?a[l]:x-1;
    for(int i=0;i<=mx;i++)
    {
        if(l==len-1&&i==0)
            continue;
        int g=ok;
        if(g) g=a[r]>=i;
        else g=a[r]>i;
        ret+=dfs(len,l-1,r+1,limit&&i==mx,g,x);
    }
    if(!limit)
        dp[len][l][x]=ret;
    return ret;
}

LL solve(int n,int x)
{
    if(n==0) return 1;
    int len=0;
    while(n){a[len++]=n%x;n/=x;}
    LL ret=1;
    for(int i=len;i>=1;i--)
        ret+=dfs(i,i-1,0,i==len,1,x);
    return ret;
}

int main()
{
    int T;scanf("%d",&T);
    memset(dp,-1,sizeof(dp));
    for(int cas=1;cas<=T;cas++)
    {
        int l,r,L,R;
        LL ans=0;
        scanf("%d%d%d%d",&L,&R,&l,&r);
        for(int i=l;i<=r;i++){
            LL t=solve(R,i)-solve(L-1,i);
            ans+=R-L+1-t+t*i;
        }
        printf("Case #%d: %lld\n",cas,ans);
    }
    return 0;
}