poj 3252(数位dp)

传送门

题解:

数位dp基础题。注意前导0的特殊处理。

P.S.空间优化:dfs函数中0和1的两个计数参数,可以二合一为cnt,表示0和1的差值,计算时整体右移32位即可(防止下标为负)

#include
#include
#include
#include
using namespace std;
int s,t,dp[33][33][33],b[33];//当前数位,当前0的出现次数,当前1的出现次数。
int dfs(int pos,int zer,int one,bool first,bool lim) {//考虑前导0
	if (pos==-1) return zer>=one;
	if (!first&&!lim&&~dp[pos][zer][one]) return dp[pos][zer][one];
	int end=lim?b[pos]:1,ans=0;
	for (int i=0;i<=end;++i) {
		if (first) {
			if (i) ans+=dfs(pos-1,0,1,0,lim&&i==end);
			else ans+=dfs(pos-1,1,0,1,lim&&i==end);
        } else ans+=dfs(pos-1,zer+(!i),one+(i&1),0,lim&&i==end);
	}
	if (!lim&&!first) dp[pos][zer][one]=ans;
	return ans;
}
inline int work(int x) {
	int len=0;
	while (x) {
		b[len++]=x&1;
		x>>=1;
	}
	return dfs(len-1,0,0,1,1);
}
int main() {
	memset(dp,-1,sizeof(dp));
	scanf("%d%d",&s,&t);
	printf("%d\n",work(t)-work(s-1));
	return 0;
}


你可能感兴趣的:(dp)