题目
题意:给定n和k,构造元素范围为
[
0
,
2
k
)
[0,2^k)
[0,2k)的元素
a
i
a_i
ai,使得
a
1
&
a
2
&
.
.
.
a
n
>
=
a
1
⨁
a
2
⨁
.
.
.
⨁
a
n
a_1\&a_2\&...a_n>=a_1 \bigoplus a_2 \bigoplus ... \bigoplus a_n
a1&a2&...an>=a1⨁a2⨁...⨁an,求有多少种构造方法。
题解:枚举每一位,dp[i][1]表示元素的前i位的与值大于异或的值,dp[i][0]表示元素的前i位的与值等于异或的值。
dp转移方程为:
dp[i][1] = dp[i][0]*quickp(2,n) + dp[i-1][1]*(全为1时,与值是否大于异或值)
dp[i][0] = dp[i-1][0]*((全为1时,与值是否等于异或值)+(偶数个1的情况数))
初始情况
dp[0][0] = (全为1时,与值是否大于异或值)
dp[0][1] = ((全为1时,与值是否等于异或值)+(偶数个1的情况数))
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn = 200010;
#define mod 1000000007
int n, k, q2, beta;
int dp[maxn][2];
int C[maxn], inv[maxn];
/*
* 0 <= a, b < mod
*/
int sub_mod(int a, int b) {
int res = a - b;
if (res < 0) res += mod;
return res;
}
/*
* 0 <= a, b < mod
*/
int add_mod(int a, int b) {
int res = a + b;
if (res >= mod) res -= mod;
return res;
}
int mul(int a, int b) {
return 1LL * a * b % mod;
}
int quick(int a, int p) {
int res = 1;
while (p) {
if (p & 1) res = mul(res, a);
a = mul(a, a);
p >>= 1;
}
return res;
}
void init() {
inv[0]=1;
inv[1]=1;
for (int i = 2; i <= n; ++i) {
inv[i] = mul(mod - mod / i, inv[mod%i]);
}
C[0] = 1;
beta = C[0];
for (int i = 1; i < n; ++i) {
// C[i] = C[i-1] * (n - i + 1) / i;
C[i] = mul(mul(C[i-1], (n - i + 1)), inv[i]);
if (i % 2 == 0) {
beta = add_mod(beta, C[i]);
}
}
q2 = quick(2, n);
}
int cal() {
if (!k) return 1;
init();
int tmp = add_mod(n & 1, beta);
dp[0][0] = (n % 2 == 0);
dp[0][1] = tmp;
for (int i = 1; i < k; ++i) {
dp[i][0] = add_mod(mul(dp[i-1][0], q2), mul(dp[i-1][1], (n % 2 == 0)));
dp[i][1] = mul(dp[i-1][1], tmp);
}
return add_mod(dp[k-1][0], dp[k-1][1]);
}
int main() {
int t;
scanf("%d", &t);
while (t--) {
scanf("%d%d", &n, &k);
printf("%d\n", cal());
}
}