2017年10月26日提高组T3 幸运值

Description

---

校庆志愿者小Z在休息时间和同学们玩卡牌游戏。一共有n张卡牌，每张卡牌上有一个数Ai，每次可以从中选出k张卡牌。一种选取方案的幸运值为这k张卡牌上数的异或和。小Z想知道所有选取方案的幸运值之和除以998244353的余数。

Input

---

输入的第一行有两个整数n和k。
第二行有n个整数，表示序列A。

Output

---

一个整数表示答案。

Hint

---

【数据规模和约定】
对于30%的数据满足，1<=n<=20
对于另30%的数据满足，1<=n<=100,0对于80%的数据满足，1<=n<=2000
对于100%的数据满足，1<=n<=100000,0

Solution

---

三十分做法显然，六十分用dp，这里写一写满分做法

题目等同于求从n个数字中取k个的异或和的和，我们知道一堆数字异或时某一位为1当且仅当这一位上共有奇数个1进行了异或操作。放到这题里面我们就可以按位来做，用组合数统计当前位异或后为1的方案数*当前位为1时的贡献就可以了

Code

---

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#define rep(i, st, ed) for (int i = st; i <= ed; i += 1)
#define drp(i, st, ed) for (int i = st; i >= ed; i -= 1)
#define erg(i, st) for (int i = ls[st]; i; i = e[i].next)
#define fill(x, t) memset(x, t, sizeof(x))
#define min(x, y) ((x)<(y)?(x):(y))
#define max(x, y) ((x)>(y)?(x):(y))
#define ll long long
#define MOD 998244353
#define INF 0x3f3f3f3f
#define N 200001
#define E 1001
#define L 1001
ll a[N], cnt[32], jc[N], r[N];
inline int read() {
    int x = 0, v = 1;
    char ch = getchar();
    for (; ch < '0' || ch > '9'; v *= (ch == '-')?(-1):(1), ch = getchar());
    for (; ch <= '9' && ch >= '0'; (x *= 10) += ch - '0', ch = getchar());
    return x * v;
}
inline ll ksm(ll x, int dep) {
    if (dep == 0) {
        return 1;
    }
    if (dep == 1) {
        return x;
    }
    ll tmp = ksm(x, dep >> 1);
    if (dep & 1) {
        return tmp * tmp % MOD * x % MOD;
    } else {
        return tmp * tmp % MOD;
    }
}
inline ll C(ll n, ll m) {
    if (n < m) {
        return 0;
    }
    return (jc[n] % MOD * r[n - m] % MOD * r[m] % MOD) % MOD;
}
int main(void) {
    int n = read();
    int k = read();
    int lim = 0;
    rep(i, 1, n) {
        a[i] = read();
        for (int tmp = a[i], j = 0; tmp; tmp >>= 1, j += 1) {
            cnt[j] += tmp & 1;
            lim = max(j, lim);
        }
    }
    jc[0] = 1;
    r[0] = 1;
    rep(i, 1, n) {
        jc[i] = (jc[i - 1] * i) % MOD;
        r[i] = ksm(jc[i], MOD - 2);
    }
    ll ans = 0;
    ll rec = 1;
    rep(i, 0, lim) {
        for (int j = 1; j <= min(k, cnt[i]); j += 2) {
            ans = (ans % MOD + rec % MOD * C(n - cnt[i], k - j) % MOD * C(cnt[i], j) % MOD) % MOD;
        }
        rec <<= 1;
    }
    std:: cout << ans << std:: endl;
    return 0;
}