poj 3252 Round Numbers

最新推荐文章于 2021-11-03 18:03:25 发布

原创最新推荐文章于 2021-11-03 18:03:25 发布 · 559 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#numbers

本文详细介绍了使用数位DP方法解决POJ3252 RoundNumbers问题的过程，包括状态定义、递归关系和边界条件的处理。通过实例分析，读者可以深入理解数位DP在处理特定类型数位问题时的优势。

poj 3252 Round Numbers

这算是个简单数位dp题目

dp[pos][zeros][ones] 表示有pos 个自由位，前面有zeros个零ones个1,的个数……要注意前导零的情况

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

int dp[32][32][32],bit[32];

int dfs(int pos,int zeros,int ones,bool ok,bool doing)
{
    if(pos<0){
         //cout<<ok<<" "<<zeros<<" "<<ones<<endl;
         if(ok) return zeros>=ones;
         else return 0;
    }
    if(!doing&&dp[pos][zeros][ones]!=-1) return dp[pos][zeros][ones];

    int end=doing?bit[pos]:1;
    int ret=0;
    for(int i=0;i<=end;i++)
    {
        int nz=zeros,no=ones;
        if(ok||i) nz=nz+(i==0),no=no+i;
        ret+=dfs(pos-1,nz,no,ok||i,doing&&i==end);
    }
    if(!doing) dp[pos][zeros][ones]=ret;
    return  ret;
 }
int solve(int n)
{
    int pos=0;
    for(;n;n>>=1,pos++)
       bit[pos]=n&1;
    return dfs(pos-1,0,0,0,1);
}
int main()
{
    memset(dp,-1,sizeof(dp));
    int l,r;
    while(scanf("%d%d",&l,&r)==2)
      printf("%d\n",solve(r)-solve(l-1));
    return 0;
}