POJ 3252 Round Numbers （数位dp）

题目：http://poj.org/problem?id=3252

题意：对于一个数，如果它的二进制表示数中，出现的0不少于1，则满足条件。求数s到e之间满足条件的数的个数。

思路：用数位dp记忆化搜索模版。一开始用sta表示0出现的次数减1出现的次数，sta会为负，数组访问时出错。

代码：

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#define MOD 1000000007
#define INF 0x7fffffff
using namespace std;
typedef long long ll;
int bit[50];
int f[50][50][50][5];

int dp(int pos, int n0, int n1, int flag, int str)
{
    //第len到pos+1位已确定，现在确定第pos位
    //n0和n1分别为0和1出现的次数
    //flag=1，不论第pos位为什么都比给定的数小
    //str=1，第len到pos+1位有不含0的数
    if(pos == 0) return n0 >= n1;
    if(flag && f[pos][n0][n1][str] != -1) return f[pos][n0][n1][str];
    int ans = 0;
    int x = flag ? 1 : bit[pos];
    for(int i = 0; i <= x; i++)
    {
        if(i == 0)
        {
            if(str) ans += dp(pos - 1, n0 + 1, n1, flag || i < x, str);
            else ans += dp(pos - 1, n0, n1, flag || i < x, str);
        }
        else
        {
            ans += dp(pos - 1, n0, n1 + 1, flag || i < x, 1);
        }
    }
    if(flag) f[pos][n0][n1][str] = ans;
    return ans;
}

int cal(int n)
{
    int len = 0;
    while(n)
    {
        bit[++len] = n % 2;
        n /= 2;
    }
    //printf("len=%d\n", len);
    int ans = dp(len, 0, 0, 0, 0);
    return ans;
}

int main()
{
    #ifdef LOCAL
    freopen("dpdata.txt", "r", stdin);
    #endif

    int n, m;
    memset(f, -1, sizeof(f));
    while(scanf("%d%d", &n, &m) != EOF)
    {
        printf("%d\n", cal(m) - cal(n - 1));
    }

    return 0;
}