@uoj - 310@ UNR #2黎明前的巧克力
Posted tiw-air-oao
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了@uoj - 310@ UNR #2黎明前的巧克力相关的知识，希望对你有一定的参考价值。
@description@
Evan 和 Lyra 都是聪明可爱的孩子，两年前，Evan 开始为一个被称为UOJ的神秘的OI组织工作，在 Evan 与其他小伙伴的努力下，UOJ不仅成了OI界原创比赛的典范，更是因UR这一反人类难度的存在而举世闻名。然而今年，随着 Evan 前往世界彼岸，UOJ一天天减少着他的活力，而就在OI历新年的黎明——NOI的前夕，刚回家不久的Evan听到了清脆的敲门声……
“开门，快递！”
“暗号？”
“3a34be41f8c8796c93b5c9f3b988e0d4”
“收到！”
送走了快递小哥，Evan 拆开包裹，看见一个写着“量子态的巧克力”的小盒子，作为新时代的OI少年，Evan 决定发扬分享的精神，于是他找来 Lyra 一起打开巧克力的盒子。
“砰——”
从量子态坍缩的光芒中回过神来，感受到脚下啫喱状的地板，Evan 叹了一口气——接着突然用把 Lyra 吓了一跳的音量喊了出来：
“竟然把造题的锅塞进巧克力里！！”
而这时，Lyra已经开始观察起了这盒奇特的巧克力，并找到了藏在巧克力盒子里的说明书。
原来，盒子里每个巧克力都有一个味道 ai，Evan 和 Lyra 的舌头上都还没有残留任何味道（用 0 表示），当他们吃下一块巧克力的时候，舌头上的味道 b 便会异或上这个巧克力的美味值，即 b ← b xor a.
Evan 和 Lyra 会各自从盒子中拿一些巧克力吃下去（即各自选择一个集合并吃掉集合中的巧克力），两个人不能吃同一块巧克力（即集合不能相交），可以有一个人选择不吃巧克力，但不能两个人都不吃。Evan 和 Lyra 不需要把盒子里的巧克力都吃完，有剩余也是可以的。
最后如果二人舌头上残留着相同的味道，则称两人是心情契合的。
既然量子态有无数种可能性，走出这巧克力迷局的关键就是，求出有多少种方案使得两人是心情契合的，两种方案不同当且仅当 Evan 或 Lyra 选择吃下的巧克力集合不一样。
你只需要输出方案数对 998244353 取模的结果即可。
@solution@
每个巧克力有三种情况：不被选中，被 E 选中，被 L 选中。条件是被 E/L 选中的巧克力异或值等于 0。
那么问题就可以转换为异或卷积：
多么优美的式子，可惜它并不能分治 fft 的套路做。
这个时候就应该考虑 fwt 的实质（我们记 c(x) 表示 x 在二进制下 1 的个数）：
可以发现 ((-1)^{c(i & a)}) 的取值只有两种，因而 fwt 中的系数也只有两种 -1, 3。稍微明朗了些。
不妨记 (cnt_i = sum_{j=1}^{n}[c(i & a_j) mod 2 = 1])，则卷积结果的 fwt 可以写作：
本道题的难点就在于求 (cnt) 了。虽然看起来很简单不过依然不好求。
解铃还须系铃人。我们尝试不绕弯路，直接使用 fwt 求上述式子。
后面那些就是 (sum_{j=1}^{n}(1 - x^{a_j})) 的 fwt 结果，因此只需一次 fwt 即可。
求出卷积结果的 fwt 后再逆 fwt 回去即可。时间复杂度 O(nlog n)。
@accepted code@
#include <cstdio>
#include <algorithm>
using namespace std;
const int MOD = 998244353;
const int MAXN = (1 << 20);
const int INV2 = (MOD + 1) / 2;
inline int add(int x, int y) {return (x + y >= MOD ? x + y - MOD : x + y);}
inline int sub(int x, int y) {return (x - y < 0 ? x - y + MOD : x - y);}
inline int mul(int x, int y) {return 1LL * x * y % MOD;}
int pow_mod(int b, int p) {
int ret = 1;
for(int i=p;i;i>>=1,b=mul(b,b))
if( i & 1 ) ret = mul(ret, b);
return ret;
}
void fwt(int *A, int n, int type) {
for(int s=2,t=1;s<=n;s<<=1,t<<=1)
for(int i=0;i<n;i+=s)
for(int j=0;j<t;j++) {
int x = A[i + j], y = A[i + j + t];
A[i + j] = add(x, y), A[i + j + t] = sub(x, y);
if( type == -1 )
A[i + j] = mul(A[i + j], INV2), A[i + j + t] = mul(A[i + j + t], INV2);
}
}
int pw3[MAXN + 5];
void init() {
pw3[0] = 1;
for(int i=1;i<=MAXN;i++)
pw3[i] = mul(3, pw3[i - 1]);
}
int f[MAXN + 5], c[MAXN + 5], a[MAXN + 5], n;
int main() {
init(), scanf("%d", &n);
for(int i=1;i<=n;i++)
scanf("%d", &a[i]), c[0] = add(c[0], 1), c[a[i]] = sub(c[a[i]], 1);
fwt(c, MAXN, 1);
for(int i=0;i<MAXN;i++) c[i] = mul(c[i], INV2);
for(int i=0;i<MAXN;i++)
f[i] = (c[i] & 1 ? sub(0, pw3[n - c[i]]) : pw3[n - c[i]]);
fwt(f, MAXN, -1);
printf("%d
", sub(f[0], 1));
}
@details@
两个集合不能同时是空集，最后方案数需要减 1。
教练：uoj 的题只有 5 页，非常小清新。大家可以尝试把它刷完。
（缓缓打出一个问号.jpg）
以上是关于@uoj - 310@ UNR #2黎明前的巧克力的主要内容，如果未能解决你的问题，请参考以下文章