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

算法分析

由题目的数据可以看到 P 是一个非常大的数,题目又涉及到了 ,那么我们不妨从二进制的角度来思考这个问题。

ex.

10 的二进制表示是 1010

21 的二进制表示是 10101

40 的二进制表示是 101000

可以发现 2x + 1 就是在二进制表示的串后添加一个 1

4x 就是在二进制表示的串后添加两个 0

那么当然有的数可以通过不同的办法同时转移得到,因此我们需要先筛选出最初始的那些数,使得所有的数都可以由初始的数转移而来。而现在我们转移的方式已经很清晰了只需要暴力枚举判断就可以了。

关于筛选的例子

比如说 3 可以由 3 本身转移而来,也可以由 2 * 1 + 1 转移而来,那么此时 3 可以变换出来的数字,一定可以通过 1 变换而来。

那么该怎么筛选呢

2x + 1 这种情况,我们可以通过 % 2 是否为 1 将其判掉

4x 这种情况,可以通过 %4 判掉

说白了就是逆推回去,看看变换前的这些初始数字是否已经在了。如果在的话说明当前的数没有价值了因为可以通过转换后得到,如果不在的话,就是我们需要的数

筛选完以后就可以通过dp的转移求解了

 : 最高位是第 i 位的,且该位上为 1 的方案数(看不懂转移方程的,建议再看看二进制转移的例子)

for(int i = 1;i <= p;i ++)
    {
        dp[i + 1] = (dp[i + 1] + dp[i]) % mod;
        dp[i + 2] = (dp[i + 2] + dp[i]) % mod;
    }

AC code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
const int N = 2e5 + 10;
vector<ll> num;
map<ll,ll> mp;
int dp[N];

bool check(int x)
{
    while(x)
    {
        if(x % 4 == 0)
        {
            x /= 4;
            if(mp[x]) return 0;
            continue;
        }
        
        if(x % 2 == 1)
        {
            x /= 2;
            if(mp[x]) return 0;
            continue;
        }
        
        break;
    }
    return 1;
}


int main()
{
    ll n,p;
    cin >> n >> p;
    for(int i = 1;i <= n;i ++)
    {
        ll x; cin >> x;
        mp[x] = 1;
        num.push_back(x);        
    }
    
    for(auto x : num)
    {
        if(check(x)) dp[__lg(x) + 1] ++;
    }
    
    for(int i = 1;i <= p;i ++)
    {
        dp[i + 1] = (dp[i + 1] + dp[i]) % mod;
        dp[i + 2] = (dp[i + 2] + dp[i]) % mod;
    }
    ll ans = 0;
    for(int i = 1;i <= p;i ++)
    ans = (ans + dp[i]) % mod;
    cout << ans << "\n";
    return 0;
}