Codeforces 1635 D. Infinite Set —— DP，斐波那契数列应用

This way

题意：

给你长度为n的数组a，然后构造一个集合，这个集合中的数存在这样的规律：
x=ai for some 1≤i≤n.
x=2y+1 and y is in S.
x=4y and y is in S.
并且x<=$2^p$，问你最终集合里有多少个数。

题解：

那么对于某个a[i]，在二进制下的情况来看，它到集合里的数可能是：
a[i]
a[i]1
a[i]00
a[i]100
a[i]001
…
可以看得出来，每次要么最后加了1，要么最后加了两个0.
dp[i]=dp[i-1]+dp[i-2]
对应两种情况
最后再做一个前缀和，于是dp[i]就表示：在数的后面增加最多i个位置的情况数。
那么不超过p位，假设a[i]有x位，a[i]的贡献就是dp[p-x]。
要注意如果当前的数能够被其它数通过题面中的操作构成，那么就不用计算了。

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
#define ll long long
const ll mod=1e9+7;
int a[N];
unordered_map<int,bool>mp;
ll dp[N];
int main()
{
    dp[1]=dp[0]=1;
    for(int i=2;i<N;i++)dp[i]=(dp[i-1]+dp[i-2])%mod;
    for(int i=1;i<N;i++)dp[i]=(dp[i-1]+dp[i])%mod;
    int n,p;
    scanf("%d%d",&n,&p);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    sort(a+1,a+1+n);
    ll ans=0;
    for(int i=1;i<=n;i++){
        int num=a[i];
        while(num){
            if(mp.count(num)){
                num=-1;
                break;
            }
            if(num&1)num/=2;
            else{
                if(num&2)break;
                num/=4;
            }
        }
        if(num==-1)continue;
        mp[a[i]]=1;
        num=0;
        while(a[i])a[i]/=2,num++;
        int res=p-num;
        if(res>=0)
            ans=(ans+dp[res])%mod;
    }
    printf("%lld\n",ans);
}