FWT模板

这个是求单点卷积值的,对于我们求某个单点的值的时候 我们并不需要完全ifwt,完全ifwt的复杂度是nlogn的,单点是on的,按照下面公式即可,其中下面的aj代表的是该数组fwt后的数组,简单理解就是fwt或卷积就是求他的子集的和,fwt与卷积就是求他超集的和,fwt异或卷积所有数都有贡献,然后看他和我们要的数的奇偶性判断正或者负贡献,偶数为正。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

//#pragma GCC target("avx")
//#pragma GCC optimize(2)
//#pragma GCC optimize(3)
//#pragma GCC optimize("Ofast")
// created by myq
#include<iostream>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<climits>
#include<cmath>
#include<cctype>
#include<stack>
#include<queue>
#include<list>
#include<vector>
#include<set>
#include<map>
#include<sstream>
#include<unordered_map>
#include<unordered_set>
using namespace std;
typedef long long ll;
#define x first
#define y second
typedef pair<int,int> pii;
const int N = 1<<17;
const int mod=998244353;
inline int read()
{
	int res=0;
	int f=1;
	char c=getchar();
	while(c>'9' ||c<'0')
	{
		if(c=='-')	f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')
	{
		res=(res<<3)+(res<<1)+c-'0';
	}
	return res;
 }
 #define int long long
 int n;
 int len;
 int a[N];
 int b[N];
  int ca[N];
 int cb[N];
 int qmi(int a,int b){
 	int res=1;
 	while(b){
 		if(b&1)	res=res*a%mod;
 		b>>=1;
 		a=a*a%mod;
	 }
	 return res;
 }
 int inv2=qmi(2,mod-2);
void fwt_xor(int f[],int inv=1)//^
{
	for(int mid=1;mid<len;mid*=2){
		for(int i=0;i<len;i+=2*mid){
			for(int k=0;k<mid;k++){
				int j=i+mid;
				int x=f[i+k];
				int y=f[j+k];
				f[i+k]=((x+y)%mod+mod)%mod;
				f[j+k]=((x-y)%mod+mod)%mod;
				if(inv==-1)
				{
					f[i+k]=f[i+k]*inv2%mod;
					f[j+k]=f[j+k]*inv2%mod;
				}
			}
			
		}
	}	
}
void fwt_and(int f[],int x=1){//&
	for(int mid=1;mid<len;mid*=2){
		for(int i=0;i<len;i+=2*mid){
			for(int k=0;k<mid;k++){
				int j=i+mid;
				f[i+k]=(f[i+k]+x*f[j+k]+mod)%mod;
			}
		}
	}	
}
void fwt_or(int f[],int x=1){//|
	for(int mid=1;mid<len;mid*=2){
		for(int i=0;i<len;i+=2*mid){
			for(int k=0;k<mid;k++){
				int j=i+mid;
				f[j+k]=(f[j+k]+f[i+k]*x+mod)%mod;
				}
			}
	}	
}
signed main()
{
	
	cin>>n;
	len=(1<<n);
	for(int i=0;i<1<<n;i++)	
		cin>>a[i];
	for(int i=0;i<1<<n;i++)
		cin>>b[i];
	memcpy(ca,a,sizeof a);
	memcpy(cb,b,sizeof b);
	fwt_or(a);
	fwt_or(b);
	for(int i=0;i<len;i++)
		a[i]=a[i]*b[i]%mod;
	fwt_or(a,-1);
	for(int i=0;i<len;i++)	cout<<a[i]<<" ";
	cout<<endl;
	memcpy(a,ca,sizeof a);
	memcpy(b,cb,sizeof b);
	fwt_and(a);
	fwt_and(b);
	for(int i=0;i<len;i++)
		a[i]=a[i]*b[i]%mod;
	fwt_and(a,-1);
	for(int i=0;i<len;i++)	cout<<a[i]<<" ";
	cout<<endl;
	memcpy(a,ca,sizeof a);
	memcpy(b,cb,sizeof b);
	fwt_xor(a);
	fwt_xor(b);
	for(int i=0;i<len;i++)
		a[i]=a[i]*b[i]%mod;
	fwt_xor(a,-1);
	for(int i=0;i<len;i++)	cout<<a[i]<<" ";
	cout<<endl;
	return 0;

}
/**
* In every life we have some trouble
* When you worry you make it double
* Don't worry,be happy.
**/



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值