Codeforces Round #772 (Div. 2)D. Infinite Set

🎉🎉🎉🎉号外号外高校算法学习社区开始新活动啦🚩🚩🚩因为同学们的基础不一样，觉得原来的每日一题比较简单,所以我们决定开设一个普及组一个提高组🙇🙇🙇提高组由我负责，每日一题，我们一起卷起来🚀🚀🚀

🌴题目描述

给你$n$个数，分别是$a_1,a_2,a_3...a_n$作为集合初始包含的数。对于$a_x(1\le x\le n)$,$4a_x$和$2a_x+1$也在集合中。现给你一个p，问你集合中小于$2^p$的数有多少个。

🌃解题报告

乘4和乘2加1这两个条件我们很容易就想到了二进制。一个操作是左移两位，一个操作是在后面添一个1嘛，看起来这两个操作好像互不影响。我们来想一下集合中若最开始只有一个数，那么他左移一位能得到一个数（1），左移两位能得到两个数（2，1+1），然后就会发现这就是爬楼梯。我们为了查询和方便，可以预处理一个前缀和。为了避免重复，就是让他一直退回去，如果这个数出现过就不去计算。

🍬AC代码（c++）

void solve() {
    mp.clear();
    fib[1]=1,fib[2]=1;
    for(int i=3;i<N;i++) (fib[i]=fib[i-1]+fib[i-2])%=MOD;
    for(int i=1;i<N;i++) (fib[i]+=fib[i-1])%=MOD;
    int n,m,ans=0;
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>q[i],mp[q[i]]=1;
    for(int i=1;i<=n;i++) {
        int res=q[i],flag=1,cnt=0;
        while(res) {
            if(res&1&&res!=1) res>>=1;
            else if(res%4==0) res>>=2;
            else break;
            if(mp[res]) {
                flag=0;
                break;
            }
        }
        if(flag==0) continue;
        while(q[i]) {
            cnt++;
            q[i]>>=1;
        }
        if(m-cnt+1<0) continue;
        (ans+=fib[m-cnt+1])%=MOD;
    }
    cout<<ans<<endl;
}