快速莫比乌斯/沃尔什变换 (FMT/FWT)

一、快速莫比乌斯变换

快速莫比乌斯变换简称 \(FMT\) ,用于快速计算位运算卷积。

我们定义 \(A\) 的莫比乌斯变换为 \(FMT(A)\)\(A_i\)\(A\) 的第 \(i\) 项。

1、或卷积

我们要求:\(\large C_x=\sum\limits_{i\cup j=x}A_iB_j\)

若有 \(\large FMT(A)_i\times FMT(B)_i=FMT(C)_i\) ,我们就能通过逆变换快速求出 \(C\) 出来。

定义:\(\large FMT(A)_n=\sum\limits_{i\in n}A_i\) ,其中 \(i,n\) 都是二进制表示的集合。

\(\large FMT(A)_x\times FMT(B)_x=\sum\limits_{i\in x}A_i\sum\limits_{j\in x}B_j=\sum\limits_{i\in x}\sum\limits_{j\in x}A_iB_j=\sum\limits_{k\in x}\sum\limits_{i\cup j=k}A_iB_j=\sum\limits_{k\in x}C_k=FMT(C)_x\)

所以利用子集和可以加速求出或卷积。

那如何快速求出子集和捏?不就是高维前缀和了捏!

高维前缀和:\(\large S_i=\sum\limits_{i\cup j=i}A_j\)

code:

void FMT(int *f,int n,int op)//op=1为正变换,op=-1为逆变换
{
    for(int i=0;i<n;++i)
        for(int j=0;j<(1<<n);++j)
            if(j&(1<<i))f[j]+=f[j^(1<<i)]*op;
}

2、与卷积

我们要求:\(\large C_x=\sum\limits_{i\cap j=x}A_iB_j\)

若有 \(\large FMT(A)_i\times FMT(B)_i=FMT(C)_i\) ,我们就能通过逆变换快速求出 \(C\) 出来。

定义:\(\large FMT(A)_n=\sum\limits_{n\in i}A_i\) ,其中 \(i,n\) 都是二进制表示的集合。

\(\large FMT(A)_x\times FMT(B)_x=\sum\limits_{x\in i}A_i\sum\limits_{x\in j}B_j=\sum\limits_{x\in i}\sum\limits_{x\in j}A_iB_j=\sum\limits_{x\in k}\sum\limits_{i\cap j=k}A_iB_j=\sum\limits_{x\in k}C_k=FMT(C)_x\)

这不就是高维后缀和了捏!

高维后缀和:\(\large S_i=\sum\limits_{i\cap j=i}A_j\)

code:

void FMT(int *f,int n,int op)//op=1为正变换,op=-1为逆变换
{
    for(int i=0;i<n;++i)
        for(int j=0;j<(1<<n);++j)
            if(j&(1<<i))f[j^(1<<i)]+=f[j]*op;
}

二、快速沃尔什变换

快速沃尔什变换简称 \(FWT\) ,同样用于快速计算位运算卷积。

我们定义 \(A\) 的沃尔什变换为 \(FWT(A)\)\(A_i\)\(A\) 的第 \(i\) 项。

1、异或卷积

定义:\(\large FWT(A)_x=\sum\limits_{k=0}^{2^n-1}(-1)^{|x\wedge k|}A_k\)

\(\large FWT(C)_x=\sum\limits_{k=0}^{2^n-1}(-1)^{|x\wedge k|}C_k\)

\(\large ~~~~~~~~~~~~~~~~~~=\sum\limits_{k=0}^{2^n-1}(-1)^{|x\wedge k|}\sum\limits_{i=0}^{2^n-1}\sum\limits_{j=0}^{2^n-1}[i\oplus j=k]A_iB_j\)

\(\large ~~~~~~~~~~~~~~~~~~=\sum\limits_{i=0}^{2^n-1}\sum\limits_{j=0}^{2^n-1}(-1)^{|(i\oplus j)\wedge x|}A_iB_j\)

\(\large ~~~~~~~~~~~~~~~~~~=\sum\limits_{i=0}^{2^n-1}\sum\limits_{j=0}^{2^n-1}(-1)^{|x\wedge i|}A_i·(-1)^{|x\wedge j|}B_j\)

\(\large ~~~~~~~~~~~~~~~~~~=(\sum\limits_{i=0}^{2^n-1}(-1)^{|x\wedge i|}A_i)(\sum\limits_{j=0}^{2^n-1}(-1)^{|x\wedge j|}B_j)\)

\(\large ~~~~~~~~~~~~~~~~~~=FWT(A)_x \times FWT(B)_x\)

和高维前缀和相似,我们对每一位依次考虑。对于第 \(i\) 位和一个不包含 \(i\) 的集合 \(S\) ,设 \(x=a_S,y=a_{S+2^i}\) ,则有新的 \(a_S=x+y,a_{S+2^i}=x-y\)

code:

void FWT(int *f,int n,int op)
{
    for(int i=1;i<(1<<n);i<<=1)
        for(int j=0;j<(1<<n);j+=(i<<1))
            for(int k=j;k<j+i;++k)
            {
                int x=f[k],y=f[k+i];
                if(op==1)f[k]=x+y,f[k+i]=x-y;
                else f[k]=(x+y)/2,f[k+i]=(x-y)/2;
            }
}

是不是很简单

当然,\(FMT\) 也可以计算或卷积和与卷积,但懒得打了(咕咕咕

2、子集卷积

upt:2022.7.7
其实半年前就学了这东西,但忘了补到博客上来了(

期望构造一个占位函数 \(w(x)\) 使得 \(i\&j=0,i|j=k\) 当且仅当 \(i\oplus j=k,w(i)+w(j)=w(k)\),容易构造 \(w(x)=|x|\)

\(f_{i,j}=\begin{cases}A_j&w(j)=i\\0&w(j)\not= i\end{cases},g_{i,j}=\begin{cases}B_j&w(j)=i\\0&w(j)\not= i\end{cases}\)

计算 \(h_k=\sum\limits_{i=0}^{k}f_i*g_{k-i}\),最后答案为 \(c_k=h_{w(k),k}\)。时间复杂度为 \(O(2_nn^2)\)

代码:

#include<bits/stdc++.h>
#define pc(x) putchar(x)
#define ll long long
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){f=ch=='-'?-1:f;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
    return x*f;
}
void write(int x)
{
    if(x<0){x=-x;putchar('-');}
    if(x>9)write(x/10);
    putchar(x%10+48);
}
const ll mod=1e9+9;
int n;
int a[21][3000005],b[21][3000005],c[21][3000005];
int cnt(int x){return __builtin_popcount(x);}
void FMT(int *f,int n,int op)
{
    for(int i=0;i<n;++i)
        for(int j=0;j<(1<<n);++j)
            if(j&(1<<i))f[j]+=f[j^(1<<i)]*op,f[j]=((ll)f[j]+mod)%mod;
}
int main()
{
    n=read();
    for(int i=0;i<(1<<n);++i)
        a[cnt(i)][i]=read();
    for(int i=0;i<(1<<n);++i)
        b[cnt(i)][i]=read();
    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<=i;++j)
            for(int k=0;k<(1<<n);++k)
                c[i][k]=(ll)(c[i][k]+(ll)a[j][k]*b[i-j][k]%mod)%mod;
    for(int i=0;i<=n;++i)FMT(c[i],n,-1);
    for(int i=0;i<(1<<n);++i)write(c[cnt(i)][i]),pc(' ');
    return 0;
}

代码是我自己打的,算法过程是搬APIO2022蔡欣然巨佬讲课的PDF。

3、子集卷积 exp

咕咕咕

4、多项式复合集合幂级数

posted @ 2022-03-24 11:45  violetctl39  阅读(729)  评论(0编辑  收藏  举报