CF #737 C. Moamen and XOR

C. Moamen and XOR

题目链接

题目大意：每组样例给一个 $n$ 和 $k$ ，问有多少个长度为 $n$ 的序列 $a$ ，其所有 $a_i$ 小于 $2^k$ ，且 $a_1\&a_2\&a_3...\&a_n\ge a_1\oplus a_2\oplus a_3...\oplus a_n$

首先考虑等于的情况，那么等于就是所有位都等于，那么我们考虑某一位相等的情况，对于这一位而言，我们要相等，又是两种情况，为 $0$ 相等或为 $1$ 相等。如果为 $0$ 相等，那么首先就是异或值要为 $0$ ，则 $n$ 个数字的这一位上的要有偶数个 $1$ ，则有 $C_n^0+C_n^2+C_n^4+...+C_n^{小于n的最大偶数}=2^{n-1}$ 。那么我们发现这里可能有种情况需要减掉，就是 $n$ 是偶数的话 $C_n^n$ 的情况，不是等于，而是大于，所以这种情况算出来之后要减一。然后考虑为 $1$ 相等的情况，那么只有全 $1$ 才可能相与为 $1$ ，但是异或那边要为 $1$ 之后奇数个 $1$ 才行，所以这种情况只能 $n$ 为奇数才能加一。我们可以最终算得某一位相等的情况为 $c$ ，所有位都相等即为 $c^k$

然后考虑大于，大于就是枚举在哪一位大于，大于只能是相与为 $1$ ，异或为 $0$ ，所以只有 $n$ 为偶数才可能发生。然后枚举在第 $i$ 位发生了大于，这一位肯定只有一种情况，就是全 $1$ 了。那么前 $i-1$ 位就要全相等，也就是有 $c^{i-1}$ 种情况，后面随便放什么都可以，那么后面一共有 $(k-i)\ast n$ 个二进制位，每个都随便放什么，所以后面的情况就是 $2^{(k-i)\ast n}$ ，所以在第 $i$ 位发生大于的情况数为： $c^{i-1}\ast 2^{(k-i)\ast n}$ ，所以大于的情况总数也就是： $\sum _{i=1}^k(c^{i-1}\ast 2^{(k-i)\ast n})$

将上述两种情况相加即可得答案。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 2e5 + 10;

const ll mod = 1e9 + 7;

ll n, k;

ll ksm(ll base, ll n) {
    ll ans = 1;
    while(n) {
        if (n & 1ll) ans = ans * base % mod;
        base = base * base % mod;
        n >>= 1ll;
    }
    return ans;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    int T;
    scanf("%d", &T);
    while(T--) {
        ll ans = 0;
        scanf("%lld%lld", &n, &k);
        ll c = ksm(2ll, n - 1);
        if (n % 2 == 0) {
            c = (c - 1 + mod) % mod;
        }
        if (n & 1) {
            c = (c + 1) % mod;
        }
        ans = (ans + ksm(c, k)) % mod;
        if (n % 2 == 0) {
            for (int i = 1; i <= k; ++i) { // 大于的情况
                ll tmp1 = ksm(c, i - 1);
                ll tmp2 = ksm(2ll, (k - i) * n);
                ans = (ans + tmp1 * tmp2 % mod) % mod;
            }
        }
        printf("%lld\n", ans);
    }
}