HDU 6156 Palindrome Function（数位DP）

 

HDU 6156 Palindrome Function

题目：

给定函数f(i,j)如果数字在j进制下为回文数的话等于j，如果不是等于1，求解

 输入：

多组输入，输入样例数t，l,r,L,R 1<=l,r<=10^9,2<=L,R,<=36

  输出：

按格式输出答案。

Sample Input

3 
1 1 2 36 
1 982180 10 10 
496690841 524639270 5 20

Sample Output

Case #1: 665 
Case #2: 1000000 
Case #3: 447525746

思路：进制数据范围比较小，枚举每一个进制。对于一个数是否为回文数，在DFS的过程使用辅助数组，和bool变量sta（是否回文），记录（len/2）前几位数，在后几位中如果与无法构成回文则，sta=false。注意前导零的存在！同时不要死板的将up在非限制时仍写成十进制下一位最大数。

#include<bits/stdc++.h>
#define ll long long
#define mod 1000000007
using namespace std;
int a[32],b[32],dp[32][32][40][2];
ll dfs(int pos,int len,int jz,bool sta,bool limit){
    if(pos==0){
        if(sta) return jz;
        return 1;
    }
    if(!limit&&~dp[pos][len][jz][sta]) return dp[pos][len][jz][sta];
    int up=limit?a[pos]:jz-1;
    ll ans=0;
    for(int i=0;i<=up;i++){
        b[pos]=i;
        if(len==pos&&i==0)
            ans+=dfs(pos-1,len-1,jz,sta,limit&&i==a[pos]);
        else if(sta&&pos<=len/2)
            ans+=dfs(pos-1,len,jz,sta&&i==b[len-pos+1],limit&&i==a[pos]);
        else ans+=dfs(pos-1,len,jz,sta,limit&&i==a[pos]);
    }
    if(!limit) dp[pos][len][jz][sta]=ans;
    return ans;
}
ll solve(ll n,int k){
    int pos=0;
    while(n){
        a[++pos]=n%k;
        n/=k;
    }
    return dfs(pos,pos,k,1,1);
}
int main()
{
    ll l,r;
    int L,R,t;
    scanf("%d",&t);
    memset(dp,-1,sizeof(dp));
    for(int s=1;s<=t;s++){
        scanf("%d%d%lld%lld",&L,&R,&l,&r);
        ll ans=0;
        for(int i=l;i<=r;i++){
            ans+= solve(R,i)-solve(L-1,i);
        }
        printf("Case #%d: %lld\n",s,ans);
    }
}