POJ 3252 Round Numbers

来源:互联网 发布:java 公章识别 编辑:程序博客网 时间:2024/05/28 23:10

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

数位DP专题:http://blog.csdn.net/chy20142109/article/details/50930004


题意:求区间内round number的个数,所谓round number就是用二进制表示时,0的个数不小于1的个数。

思路:记录从最高层dfs到当前层的0的个数,1的个数,以及是否用1开过头了,还是前面选的全是0。
如果前面全是0,可以继续选0或者在当前层以1开头。或者就当前层选1/0。


#include <cstdio>#include <cmath>#include <cstring>#include <string>#include <cstdlib>#include <iostream>#include <algorithm>#include <stack>#include <map>#include <set>#include <vector>#include <sstream>#include <queue>#include <utility>using namespace std;#define rep(i,j,k) for (int i=j;i<=k;i++)#define Rrep(i,j,k) for (int i=j;i>=k;i--)#define Clean(x,y) memset(x,y,sizeof(x))#define LL long long#define ULL unsigned long long#define inf 0x7fffffff#define mod %100000007int dp[35][35][35];int bit[35];int st,ed;int dfs(int k , int zero , int one , bool f , bool flag){    if ( k == 0 ) return ( f && zero>=one );    if ( !flag && dp[k][zero][one] != -1 ) return dp[k][zero][one];    int ed = flag?bit[k]:1;    int ans = 0;    rep(i,0,ed)    {        if( f )            ans+=(i==0)?dfs( k - 1 , zero + 1 , one , f , flag && i == ed ):dfs( k - 1 , zero , one + 1 , f , flag && i == ed );        else ans+=(i==0)?dfs( k - 1 , 0 , 0 , 0 , 0 ):dfs( k - 1 , 0 , 1 , 1 , flag && i == ed );    }    if ( !flag ) dp[k][zero][one] = ans;    return ans;}int cal(int x){    int len = 0;    while(x)    {        bit[++len] = x % 2;        x /= 2;    }    return dfs(len,0,0,0,1);}int main(){    Clean(dp,-1);    cin>>st>>ed;    cout<<cal(ed)-cal(st-1);    return 0;}


0 0
原创粉丝点击