快速莫比乌斯变换

莫比乌斯变换

定义函数$f$的莫比乌斯变换为$\hat{f}$
$${\hat{f}}_S=\sum _{T\subseteq S}{f}_T$$
则有莫比乌斯反演
$$f_S=\sum _{T\subseteq S}(-1{)}^{|S|-|T|}{\hat{f}}_T$$

快速莫比乌斯变换FMT

考虑如何快速快速进行莫比乌斯变换和反演。

• 设 f ^ S i = ∑ T ⊆ S [ ( S − T ) ⊆ { 1 , 2 , . . . , i } ] f T \hat f_S^i=\sum\limits_{T \subseteq S} [(S - T) \subseteq \{1, 2, ..., i\}] f_{T} f^​Si​=T⊆S∑​[(S−T)⊆{1,2,...,i}]fT​
• 易得初始状态：${\hat{f}}_S^0=f_S$， { 1 , 2..0 } = ∅ \{1,2..0\}=\empty {1,2..0}=∅
• 对于每一个不包含 { i } \{i\} {i}的集合$S$，可知 ${\hat{f}}_S^i={\hat{f}}_S^{i-1}$

f ^ S ∪ { i } i = ∑ T ⊆ S ∪ { i } [ ( S ∪ { i } − T ) ⊆ { 1 , 2 , . . . , i } ] f T \hat f_{S \cup \{i\}}^i=\sum\limits_{T \subseteq S\cup\{i\}} [(S\cup\{i\} - T) \subseteq \{1, 2, ..., i\}] f_{T} f^​S∪{i}i​=T⊆S∪{i}∑​[(S∪{i}−T)⊆{1,2,...,i}]fT​
其中$T$可以分为一定包含$i$和一定不包含$i$两种情况。

• 第一种情况 ( S ∪ { i } − T ) (S\cup\{i\} - T) (S∪{i}−T)一定不包含$i$，所以
  [ ( S ∪ { i } − T ) ⊆ { 1 , 2 , . . . , i } ] ⇔ [ ( S ∪ { i } − T ) ⊆ { 1 , 2 , . . . , i − 1 } ] [(S\cup\{i\} - T) \subseteq \{1, 2, ..., i\}]\Leftrightarrow[(S\cup\{i\} - T) \subseteq \{1, 2, ..., i-1\}] [(S∪{i}−T)⊆{1,2,...,i}]⇔[(S∪{i}−T)⊆{1,2,...,i−1}]
  所以这部分的贡献为 f ^ S ∪ { i } i − 1 = ∑ T ⊆ S ∪ { i } [ ( S ∪ { i } − T ) ⊆ { 1 , 2 , . . . , i − 1 } ] f T \hat f_{S\cup\{i\}}^{i-1}=\sum\limits_{T\subseteq S\cup \{i\}}[(S\cup\{i\} - T) \subseteq \{1, 2, ..., i-1\}]f_T f^​S∪{i}i−1​=T⊆S∪{i}∑​[(S∪{i}−T)⊆{1,2,...,i−1}]fT​
• 第二种情况
  T ⊆ S ∪ { i } ⇔ T ⊆ S [ ( S ∪ { i } − T ) ⊆ { 1 , 2 , . . . , i } ] [ ( S ∪ { i } − T − { i } ) ⊆ { 1 , 2 , . . . , i } − { i } ] [ ( S − T ) ⊆ { 1 , 2 , . . . , i − 1 } ] \begin{aligned} T\subseteq S\cup\{i\}&\Leftrightarrow T\subseteq S\\ [(S\cup\{i\} - T)&\subseteq \{1, 2, ..., i\}]\\ [(S\cup\{i\} - T-\{i\})&\subseteq \{1, 2, ..., i\}-\{i\}]\\ [(S - T)&\subseteq \{1, 2, ..., i-1\}] \end{aligned} T⊆S∪{i}[(S∪{i}−T)[(S∪{i}−T−{i})[(S−T)​⇔T⊆S⊆{1,2,...,i}]⊆{1,2,...,i}−{i}]⊆{1,2,...,i−1}]​
  所以这部分贡献为 f ^ S i − 1 = ∑ T ⊆ S [ ( S − T ) ⊆ { 1 , 2 , . . . , i − 1 } ] f T \hat f_S^{i-1}=\sum\limits_{T\subseteq S}[(S - T)\subseteq \{1, 2, ..., i-1\}]f_T f^​Si−1​=T⊆S∑​[(S−T)⊆{1,2,...,i−1}]fT​

综上
f ^ S ∪ { i } i = f ^ S i − 1 + f ^ S ∪ { i } i − 1 \hat f_{S\cup\{i\}}^i=\hat f_S^{i-1}+\hat f_{S\cup\{i\}}^{i-1} f^​S∪{i}i​=f^​Si−1​+f^​S∪{i}i−1​
$O(n{2}^n)$递推即可，反演也可以用类似的方法分析，最后得到
f S ∪ { i } i = f S ∪ { i } i − 1 − f S i − 1 f_{S\cup\{i\}}^i=f_{S\cup\{i\}}^{i-1}-f_S^{i-1} fS∪{i}i​=fS∪{i}i−1​−fSi−1​

void FMT(int *a,int n,int flag) {
    for(int i=0; i<n; i++)
        for(int j=0; j<(1<<n); j++)
            if(j>>i&1)
                a[j]=(a[j]+(flag+mod)%mod*a[j^(1<<i)]%mod)%mod;
}

应用

1. 求集合的卷积$h=f\ast g$，$h_S=\sum _{L\cup R=S}{f}_L\times g_R$
   $$\begin{array}{rl}{\hat{h}}_S& =\sum _{T\subseteq S}{h}_T\\ & =\sum _{T\subseteq S}\sum _{L\cup R=T}{f}_L\times g_R\\ & =\sum _{L\subseteq S}\sum _{R\subseteq S}{f}_L\times g_R\\ & =\sum _{L\subseteq S}{f}_L\sum _{R\subseteq S}{g}_R\\ & ={\hat{f}}_S\times {\hat{g}}_S\end{array}$$
   即
   $$FMT(h)=FMT(f)\times FMT(g)$$
   先对$f$和$g$进行快速莫比乌斯变换，然后对应项相乘，得到$h$的莫比乌斯变换，然后进行莫比乌斯逆变换(反演)，得到$h$。形式和快速傅里叶变换一样。

2. 另一种集合的卷积$h_S=\sum _{L\cup R=S,L\cap R=\varnothing }{f}_L\times g_R$
   令$F_{i,j}=\{\begin{array}{ll}{f}_{i,j}& |j|=i\\ 0& |j|\ne i\end{array}$ $i$是元素的个数，$j$是集合，$|j|$是集合$j$的大小，同理$G_{i,j}$，那么$L\cap R=\varnothing \iff |L|+|R|=|S|$则有
   $$\begin{array}{rl}{H}_{i,j}& =\sum _{i_1+i_2=i}\sum _{L\cup R=j}{F}_{i_1,L}\times G_{i_2,R}\\ H_i& =\sum _{i_1+i_2=i}{F}_{i_1}\ast G_{i_2}(省略第二维)\\ FMT(H_i)& =\sum _{i_1+i_2=i}FMT(F_{i_1})\times FMT(G_{i_2})\end{array}$$
   最后要的答案就是满足$|j|=i$的$H_{i,j}$
   例题

#include<bits/stdc++.h>
#define ll long long
#define mod 1000000009
using namespace std;
const int N=22;
int n,len,a[N][1<<N],b[N][1<<N],h[N][1<<N];
int ct(int x){
    return __builtin_popcount(x);
}
void FMT(int *a,int n,int flag) {
    for(int i=0; i<n; i++)
        for(int j=0; j<(1<<n); j++)
            if(j>>i&1)
                a[j]=(a[j]+(flag+mod)%mod*1ll*a[j^(1<<i)]%mod)%mod;
}
int main() {
    scanf("%d",&n);
    len=1<<n;
    for(int i=0;i<len;i++)
        scanf("%d",&a[ct(i)][i]);
    for(int i=0;i<len;i++)
        scanf("%d",&b[ct(i)][i]);
    for(int i=0;i<=n;i++)
        FMT(a[i],n,1),FMT(b[i],n,1);
    for(int i=0;i<=n;i++){
        for(int j=0;j<len;j++)
        for(int k=0;k<=i;k++)
        h[i][j]=(h[i][j]+1ll*a[k][j]*b[i-k][j]%mod)%mod;
        FMT(h[i],n,-1);
    }
    for(int i=0;i<len;i++)
        printf("%d ",h[ct(i)][i]);
    return 0;
}