hdu4734 F(x)（数位dp）

hdu4734

题目

给了F(x) = An * 2n-1 + An-1 * 2n-2 + … + A2 * 2 + A1 * 1，现给出A,B求0~B之间有多少数字的F小于F（A）

思路

就是数位dp，记一位状态sum表示要小于的F值，然后一位位向下推，sum一次次的减。。。

代码

#include
#include
#include
#include
#include
#include
#include

using namespace std;

typedef  long long ll;
int dp[11][100000];
int c[11];

int disb[11];

int dfs(int len,int sum,int lim)
{
    if(sum<0) return 0;
    if(len==-1) return sum>=0;
    if(!lim&&dp[len][sum]!=-1) return dp[len][sum];
    int res=0;
    int u=lim?disb[len]:9;
    for(int i=0;i<=u;i++)
    {
        res+=dfs(len-1,sum-i*c[len],lim&&i==u);
    }
    if(lim) return res;
    else return dp[len][sum]=res;

}

int solve(int x,int y)
{
    int ans=0;
    int i=0;
    while(y)
    {
        ans+=c[i++]*(y%10);
        y/=10;
    }
    int  len=0;
    while(x)
    {
        disb[len++]=x%10;
        x/=10;
    }
    return dfs(len-1,ans,1);
}

int a,b;

int main()
{
    int T;
    scanf("%d",&T);
    int kase=1;
    for(int i=0; i<=10; i++)
        c[i]=1<memset(dp,-1,sizeof(dp));
    while(T--)
    {
        scanf("%d %d",&a,&b);
        printf("Case #%d: %d\n",kase++,solve(b,a));
    }
}