【CF1635D】Infinite Set【DP】【位运算】

分析

看到 $2^p$ 如此之大，应该要往位运算上想。而且2x+1和4x也具有明显的位运算特征：一个是在后面加上“1”，一个是加“00”。

列方程显然可得：2x+1与4x生成的数互相不同不会干扰。

根据这个递推式可以知道对于每个答案就类似于斐波那契数列前n项和（做一个前缀和），但是要根据剩余可行位数（用log就可以算出）来计算。

但是此处有个问题，假设a数组中出现5 20，实际上20是没用的，那怎么去掉这些没用的（可以被表示的）数呢？直接用一个int->bool的映射，从小到大排序之后，对于每个数要不去掉1要不去掉00，直到不能再操作而又没有遇到已经操作成功的数就行。

思路是没问题的，然而，第九个点一直过不去不知道为什么，代码先贴这了。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cmath>
#include<cstdlib>
using namespace std;
typedef long long ll;

map<int,bool> v;
const int mod=1e9+7;
int n,p,a[200010],b[200010],tot;
ll f[200010],s[200010],ans;

bool check(int x)
{
	while(x)
	{
		if(v[x]) return false;
		if(x&1) x>>=1;
	    else if(x&3) break;
	    else x>>=2;
	}
	return true;
}

int main()
{
	cin>>n>>p;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	sort(a+1,a+n+1);
	for(int i=1;i<=n;i++)
	{
		if(check(a[i])) v[a[i]]=true,b[++tot]=a[i];//标记可保留 
	}
	f[0]=f[1]=1;
	for(int i=2;i<=p+1;i++) f[i]=(f[i-1]+f[i-2])%mod;//递推 
	s[0]=1;
	for(int i=1;i<=p+1;i++) s[i]=(s[i-1]+f[i])%mod;//前缀和 
	for(int i=1;i<=tot;i++)
	{
		ans=(ans+s[p-(int(log2(b[i]))+1)])%mod;
	}
	cout<<ans%mod;
	return 0;
}