好数

好数

题目大意

对于一个正整数X，如果把X化成二进制数后，如果X的二进制数至少有三个连续的1或者至少有3个连续的0（不能有前导0），那么X就是“好数”。
现在给出两个整数Low、UP，求Low和UP范围内有多少个“好数”。

输入样例

0 16

输出样例

5

提示

思路

这道题，要用数位dp来做。
因为求好数比较难，那我们不如反着求，求“坏数”。（既是没有三个连续且一样的数）

先用数位dp分位数求出每一位用什么开头的坏数个数，既是其位数的最小值和最大值之间有多少坏数。
把1位到二进制的位数的每一位的坏数个数加起来，就是1位到最后一位的坏数总数。

但是，最后一位并不一定是到最大值。所以，对于最后一位，我们要用另外一种方法。
我们可以用已经求出来的坏数总数，分为不同位数的相加，就可以了。

代码

#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
ll aa,bb,a[41],f[40][5][5];
ll work(ll wei)
{
	ll ans=1;
	for (int i=wei-1;i>0;i--)
	ans+=f[i][1][0]+f[i][1][1];//把坏数个数加在一起
	for (int i=wei-1;i>0;i--)//最后一位（特殊情况）
	{
	  	if (a[i]==1)
		if (a[i+1]==1||a[i+2]==1||i==wei-1)//改1为0
		ans+=f[i+1][a[i+1]][0];//相加
		if (a[i]==a[i+1]&&a[i+1]==a[i+2]&&i!=wei-1)//判断是否变成好数
		{
		  	ans--;
		  	break;
		}
	}
	return ans;
}
ll er(ll dep)
{
	if (dep<=0) return 0;
	ll t=dep,wei=0;
	memset(a,0,sizeof(a));//初始化
	while (t)//转二进制
	{
		a[++wei]=t&1;
		t>>=1;
	}
	return dep-work(wei);//因为要的是好数，所以要用总数减它
}
int main()
{
//	freopen("d.in","r",stdin);
//	freopen("d.out","w",stdout);
	f[1][1][0]=1;f[1][0][1]=1;//数位dp初始化
	for (int k=2;k<=35;k++)//数位dp
	{
		f[k][0][0]=f[k-1][0][1];
		f[k][0][1]=f[k-1][1][1]+f[k-1][1][0];
		f[k][1][0]=f[k-1][0][0]+f[k-1][0][1];
		f[k][1][1]=f[k-1][1][0];
	}
	scanf("%lld%lld",&aa,&bb);//输入
	printf("%lld",er(bb)-er(aa-1));//输出
//	fclose(stdin);
//	fclose(stdout);
}