POJ-3252 Round Number(数位DP)

题意,给定一个区间[a,b],求满足二进制位零的个数大于等于一的个数的数的个数。(0<a<b<2e9)

裸的数位DP,状态保存零的个数减一的个数,由于会出现负数,再加上32。

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
using namespace std;
int num[33];
int dp[33][67];

int dfs(int k,int status,bool ismax)
{
	if(k==0)return status>=32;
	if(!ismax && dp[k][status]!=-1)return dp[k][status];
	int maxer=ismax?num[k]:1,res=0;
	FOR(i,0,maxer)
	{
		if(status==65)res+=dfs(k-1,i?31:65,ismax&&i==maxer);
		else res+=dfs(k-1,status+((i==0)<<1)-1,ismax&&i==maxer);
	}
	if(!ismax)dp[k][status]=res;
	return res;
}
int solve(int k)
{
	int p=0;
	while(k)
	{
		num[++p]=(k&1);
		k>>=1;
	}
	return dfs(p,65,1);
}

int main()
{
	int A,B;
	memset(dp,-1,sizeof(dp));
	scanf("%d%d",&A,&B);
	printf("%d\n",solve(B)-solve(A-1));
	return 0;
}

猜你喜欢

转载自blog.csdn.net/paulliant/article/details/80359035