poj3252 Round Numbers（数位dp）

poj3252

题目

就是求指定范围内转换成二进制0的数目大于等于一的数目的数的个数。

思路

dp[pos][no][nz],pos表示第几位，no表示零的个数，nz表示1的个数，转移的时候我多加了一个first判断有没有前导零，分了情况讨论。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
#include<stack>

using namespace std;

typedef  long long ll;
ll dp[40][40][40];
int dis[40];


ll dfs(int len,int first,int no,int nz,int lim)
{
    if(len<0) return first||nz>=no;
    if(!first&&!lim&&dp[len][no][nz]!=-1) return  dp[len][no][nz];
    ll res=0;
    int u=lim?dis[len]:1;

    for(int i=0; i<=u; i++)
        if(first)
        {
            if(i==0)
                res+=dfs(len-1,1,no,nz,lim&&i==u);
            else
                res+=dfs(len-1,0,no+1,nz,lim&&i==u);
        }
        else
        {
            if(i==0)
                res+=dfs(len-1,0,no,nz+1,lim&&i==u);
            else
                res+=dfs(len-1,0,no+1,nz,lim&&i==u);
        }
    if(!lim&&!first) return dp[len][no][nz]=res;
    else return res;
}

ll solve(ll x)
{
    int len=0;
    while(x)
    {
        dis[len++]=x%2;
        x/=2;
    }
    return dfs(len-1,1,0,0,1);
}

ll a,b;

int main()
{
    scanf("%I64d %I64d",&a,&b);
    memset(dp,-1,sizeof(dp));
    printf("%I64d\n",solve(b)-solve(a-1));
}