poj3252 Round Numbers

Posted 2021-01-09 gaudar

题意：一个数，二进制形式去掉leading zero后，如果0的数量大于等于1的数量，计算区间内这样数的个数。

右区间转为二进制形式，然后在二进制上数位dp,递归的时候记录leading zero是否消除了，以及0比1多的数量delta。dp记录时有没有leading zero也是一个状态。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cstring>
#include <map>
#include <queue>
#include <set>
#include <cassert>
using namespace std;
const double EPS=1e-8;
const int SZ=5050,INF=0x7FFFFFFF;
typedef long long lon;
int dp[50][120][2];

int dfs(int pos,int delta,int limit,string &str,int lead)
{
    if(pos==str.size())return delta>=0&&!lead;
    if(!limit&&dp[str.size()-pos-1][delta+60][lead]!=-1)return dp[str.size()-pos-1][delta+60][lead];
    int up=limit?str[pos]-‘0‘:1;
    int res=0;
    for(int i=0;i<=up;++i)
    {
        if(!i)
        {
            if(lead)res+=dfs(pos+1,delta,limit&&str[pos]-‘0‘==i,str,1);
            else res+=dfs(pos+1,delta+1,limit&&str[pos]-‘0‘==i,str,0);
        }
        else res+=dfs(pos+1,delta-1,limit&&str[pos]-‘0‘==i,str,0);
    }//一直将str[pos]写成str[i] 
    if(!limit)dp[str.size()-pos-1][delta+60][lead]=res;
    return res;
}

int work(string &str)
{
    return dfs(0,0,1,str,1);
}

string parseBinary(int x)
{
    string res="";
    for(int i=0;i<32;++i)
    {
        if(x&(1<<i))res="1"+res;
        else res="0"+res;
    }
    res.erase(0,res.find_first_not_of(‘0‘));///??????empty
    return res;
}

int main()
{
    std::ios::sync_with_stdio(0);
    //freopen("d:\1.txt","r",stdin);
    //lon casenum;
    //cin>>casenum;
    memset(dp,-1,sizeof(dp));
    int ll,rr;
    //for(lon time=1;time<=casenum;++time)
    //for(lon time=1;cin>>ll>>rr;++time)
    {
        cin>>ll>>rr;
        --ll;
        string sll=parseBinary(ll),srr=parseBinary(rr);
        cout<<work(srr)-work(sll)<<endl;
    }
    return 0;
}