组合计数 - Moamen and XOR - CF 1557C

组合计数 - Moamen and XOR - CF 1557C

题意：

$给定一个长度为n的非负整数序列a，其中每个元素均小于2^k$

$要求满足：$

$$a_1\&a_2\&a_3\&...\&a_n\ge a_1\oplus a_2\oplus a_3\oplus ...\oplus a_n$$

$计算满足上述条件的序列a的数量，答案对10^9+7取模。$

输入：

$多测，1\le T\le 5$

$每组测试数据包括：n和k(1\le n\le 2\cdot 10^5，1\le k\le 2\cdot 10^5)$

输出：

$一个整数，表示答案$

Example
input

3
3 1
2 1
4 0

output

5
2
1

---

结论：

$①、若n为奇数：$

$$Ans=(2^{n-1}+1{)}^k$$

$②、若n为偶数：$

$$Ans=(2^{n-1}-1{)}^k+\sum _{i=1}^k(2^{n-1}-1{)}^{i-1}\cdot (2^n{)}^{k-i}$$

分析：

题意本质上是求：选择n个k位二进制数，满足上述约束，计算合法方案总数。

因为二进制的每一位都是独立的，所以我们无须纠结具体的每一个数，仅需从运算结果入手分析，

单独分析结果每一位，根据乘法原理，最后将每一位的合法方案数累乘即可。

$记L={\&}_{i=1}^n{a}_i，R={\oplus }_{i=1}^n{a}_i$

$记L的二进制表示，第i位数为L_i；同理有R_i。$

$记a_j的二进制表示，第i位数为a_{ji}。$

$我们粗略的分析：$

$若对于1\le i\le m，L_i=R_i，$

$当L_{i+1}=R_{i+1}时，L才可能等于R；$

$当L_{i+1}>R_{i+1}时，L必大于R。$

$又因为，结果的每一位都是独立的，而结果的每一位和1的个数的奇偶性相关，所以下面分奇偶讨论：$

$①、当n为奇数$

$$\begin{gathered} 若存在a_{ji}=0(1\le i\le k)，则L_i=0，L_i\le R_i； \\ 若L_i=1，奇数个1异或仍为1，即R_i=1，此时L_i=R_i。 \end{gathered}$$

$可见：当n为奇数时，L_i只可能等于R_i，即L只能等于R。下面我们详细讨论每一位的值：$

$Ⅰ、L_i=R_i=1，此时仅有1种情况：\forall j\in [1,n]，a_{ji}=1$

$Ⅱ、L_i=R_i=0，此时我们需要选择偶数个a_{ji}=1，使得异或值为0，故方案数为：$
$$C_n^0+C_n^2+C_n^4+...+C_n^{n-1}=C_n^1+C_n^3+...+C_n^{n-2}+C_n^n=\frac{2^n}{2}=2^{n-1}$$

$所以，当n为奇数时，Ans={\prod }_{i=1}^k(2^{n-1}+1)=(2^{n-1}+1{)}^k。$

$②、当n为偶数$

$类比情况①，我们可以分析出合法方案为：L\ge R$

$Ⅰ、左边等于右边，只可能有L_i=R_i=0，此时为偶数个1(且存在a_{ji}=0)，即不全为1$

$第i位相等的合法方案数为：$

$$C_n^0+C_n^2+C_n^4+...+C_n^{n-2}=C_n^0+C_n^2+C_n^4+...+C_n^{n-2}+C_n^n-C_n^n=2^{n-1}-1$$

$注：二项式系数和的性质，S_{even}=S_{odd}=2^{n-1}$

$Ⅱ、左边大于右边$

$注意，L>R无须任意的L_i>R_i，我们仅需保证前i-1位L_{i-1}=R_{i-1}，第i位L_i=1且R_i=0即可。$

$根情况Ⅰ，前i-1位相等的合法方案数为：(2^{n-1}-1{)}^{i-1}；第i位向后任意，方案数为：(2^n{)}^{k-i}$

$故合法方案数为：$

$$\sum _{i=1}^k(2^{n-1}-1{)}^{i-1}\cdot (2^n{)}^{k-i}$$

$故n为偶数时，合法方案总数为：$$Ans=(2^{n-1}-1{)}^k+{\sum }_{i=1}^k(2^{n-1}-1{)}^{i-1}\cdot (2^n{)}^{k-i}$

代码：

#include<cstdio>
#include<algorithm>
#include<iostream>

using namespace std;

const int N = 200010, mod = 1e9 + 7;
typedef long long ll;

int T, n, k;
ll ans, pow_2[N];

void init()
{
	pow_2[0] = 1;
	for(int i = 1; i < N; i ++)
		pow_2[i] = pow_2[i - 1] * 2 % mod;
}

ll quick_pow(int a, int b)
{
	ll res = 1;
	while(b)
	{
		if(b & 1) res = res * a % mod;
		a = (ll)a * a % mod;
		b >>= 1;
	}
	return res;
}

int main()
{
	init();
	scanf("%d", &T);
	while(T --)
	{
		scanf("%d%d", &n, &k);
		ll ans = 0;
		if(n & 1) ans = quick_pow(pow_2[n - 1] + 1, k);
		else
		{
			ans = quick_pow(pow_2[n - 1] - 1, k);
			ll sum = 0; 
			for(int i = 1; i <= k; i ++)
				sum = (sum + quick_pow(pow_2[n - 1] - 1, i - 1) * quick_pow(pow_2[n], k - i) % mod) % mod;
			ans = (ans + sum) % mod;
		}
		printf("%lld\n", ans);
	}
	
	return 0;
}