给定
N
(
≤
5
×
1
0
5
)
N(≤5\times10^5)
N(≤5×105)个长度为20的01串,每次可选择其中恰好
K
K
K个。
Q
(
≤
5
×
1
0
5
)
Q(≤5\times10^5)
Q(≤5×105)个询问,每个询问表示为一个长度为20的01串,若其某一位为1则选择的
K
K
K个串中至少要有一个的该位为1。求方案数。
Solution
考虑容斥。设
f
[
i
]
f[i]
f[i]表示选择了
K
K
K个与
i
i
i按位与的答案均为0的串的方案数,则根据二项式反演易得:
a
n
s
S
=
∑
i
⊂
S
(
−
1
)
∣
i
∣
f
[
i
]
ans_S=\sum_{i\subset S} (-1)^{|i|}f[i]
ansS=∑i⊂S(−1)∣i∣f[i]。
我们再设
g
[
i
]
g[i]
g[i]表示满足与
i
i
i按位与的答案均为0的串的个数,则显然
f
[
i
]
=
C
(
g
[
i
]
,
K
)
f[i]=C(g[i],K)
f[i]=C(g[i],K)。
现在,考虑快速求出
g
[
i
]
g[i]
g[i]和
a
n
s
S
ans_S
ansS。
对于
N
N
N个串中的任意一个,我们可以先将其取反,然后再令它的每个子集的
g
[
i
]
g[i]
g[i]++。
如果反过来,令所有包含某个小集合的大集合的
g
[
i
]
g[i]
g[i]++,那就相当于求一个高维前缀和(每位1包含0的答案);而现在则可以类比为高维后缀和(每位0包含1的答案)。前者的实现是按位合并;后者的实现也差不多(详见Code)。
而
a
n
s
S
=
∑
i
⊂
S
(
−
1
)
∣
i
∣
C
(
g
[
i
]
,
K
)
ans_S=\sum_{i\subset S} (-1)^{|i|}C(g[i],K)
ansS=∑i⊂S(−1)∣i∣C(g[i],K),则相当于求高维前缀和,另开一个数组存储即可。