hdu4734（数位dp）

hdu4734

给定 a和b，

问区间[0,b]内有多少个数字的f(i) <=f(a)

dp[i][s] 表示i位的数字的f<=s

所以比如如果第i+1位选择数字5之后， 那么只要剩下的i位数字的f<=f(a) - 5*2^i 即可。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <string>
#include <math.h>
using namespace std;
#pragma warning(disable:4996)
#pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long LL;                   
const int INF = 1<<30;
/*
4599
满足一个性质， 但是这个性质是会变化的
*/
int num[12];
int bit[12];
int dp[12][5555];
void init()
{
    bit[0] = 1;
    for (int i = 1; i <= 10; ++i)
        bit[i] = bit[i - 1] * 2;
}



int dfs2(int pos, int sta, bool flag)
{
    if (sta < 0) return 0;
    if (pos == 0) return 1;
    if (!flag && dp[pos][sta] != -1)
        return dp[pos][sta];
    int end = flag ? num[pos] : 9;
    int ans = 0;
    for (int i = 0; i <= end; ++i)
    {
        ans += dfs2(pos - 1, sta - bit[pos - 1] * i, flag&&i == end);
    }
    if (!flag)
        dp[pos][sta] = ans;
    return ans;
}
int calc(int sta, int n)
{
    int len = 0;
    while (n)
    {
        num[++len] = n % 10;
        n /= 10;
    }
    return dfs2(len, sta, true);
}
int main()
{
    init();
    memset(dp, -1, sizeof(dp));
    int t,a,b;
    scanf("%d", &t);
    for (int k = 1; k <= t; ++k)
    {
        scanf("%d%d", &a, &b);
        int len = 0, sta = 0;
        while (a)
        {
            num[++len] = a % 10;
            a /= 10;
        }
        for (int i = len; i >= 1; --i)
            sta += bit[i - 1] * num[i];
        printf("Case #%d: %d
",k, calc(sta,b));
    }
    return 0;
}