码迷,mamicode.com
首页 > 其他好文 > 详细

UOJ #310 黎明前的巧克力 (FWT)

时间:2019-03-25 23:20:40      阅读:205      评论:0      收藏:0      [点我收藏+]

标签:string   异或   其它   ace   函数   return   pac   算法理解   can   

题目传送门

题目大意:给你一个序列,定义一个子序列的权值表示子序列中元素的异或和,现在让你选出两个互不相交的子序列,求选出的这两个子序列权值相等的方案数,n,a_{i}\leq 10^{6}

这是一道考察对FWT算法理解的好题。然而我并不会

思路来自出题人的题解

 

假设权值最大值为$m$

暴力怎么搞?背包$DP$一下

定义$f(i,j)$表示现在遍历到了第$i$个元素,选出的两个子序列异或和为$j$的方案数,容易得到方程:

$f(i,j)=f(i-1,j)+2*f(i-1,j\;xor\;a_{i})$

时间复杂度$O(nm)$,可以获得30分

 

看那个卷积形式,$FWT$?

时间复杂度$O(nmlogm)$,可以获得..0分

 

我们发现每一层的生成函数里只有两个位置有值

假设现在我们遍历到了第$i$个物品$a_{i}$,第$i$层的生成函数长这个样:

$f_{i}(0)=1, f_{i}(a_{i})=2$

其它位置都是0诶

对它进行FWT正变换,会发现正变换之后的数组里只有-1和3

因为$f_{i}(0)$对每个位置都有+1的贡献,而$f_{i}(a_{i})$对每个位置有+2或-2点贡献

重新考虑那个$0$分暴力。我们把每一层都正变换,然后对应位置相乘,再逆变换回来

我们可以想办法快速求出对应位置相乘之后的数组$F$,这样再用一次逆变换就行了

我们只需要统计出每个位置上有几个3相乘,设3的数量是$x$,-1的数量就是$n-x$,快速幂一下,就能得到$F$了

我们把贡献积转化成了指数上的贡献和,发现只用一次正变换就行啦!

再用快速幂把贡献和转化成贡献积。最后逆变换回来就行了

时间$O(mlogm)$

 

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #def_{i}ne N1 (1<<20)+10
 5 #def_{i}ne ll long long 
 6 using namespace std;
 7 const int p=998244353;
 8 
 9 template <typename _T> void read(_T &ret)
10 {
11     ret=0; _T fh=1; char c=getchar();
12     while(c<0||c>9){ if(c==-) fh=-1; c=getchar(); }
13     while(c>=0&&c<=9){ ret=ret*10+c-0; c=getchar(); }
14     ret=ret*fh;
15 }
16 
17 ll qpow(ll x,ll y)
18 {
19     ll ans=1;
20     for(;y>0;x=x*x%p,y>>=1) if(y&1) ans=ans*x%p;
21     return ans;
22 }
23 
24 void FWT_XOR(int *s,int len,int type)
25 {
26     int i,j,k,t,inv2=qpow(2,p-2);
27     for(k=2;k<=len;k<<=1)
28     for(i=0;i<len;i+=k)
29     for(j=0;j<(k>>1);j++)
30     {
31         t=s[i+j+(k>>1)]; s[i+j+(k>>1)]=(s[i+j]-t+p)%p; s[i+j]=(s[i+j]+t)%p;
32         if(type==-1) s[i+j]=1ll*s[i+j]*inv2%p, s[i+j+(k>>1)]=1ll*s[i+j+(k>>1)]*inv2%p;
33     }
34 }
35 
36 int n,ma,len,L;
37 int a[N1],s[N1];
38 
39 int ma_{i}n()
40 {
41     scanf("%d",&n);
42     int i,j,x;
43     for(i=1;i<=n;i++) read(a[i]);
44     for(i=1;i<=n;i++) s[a[i]]++, ma=max(ma,a[i]);
45     for(len=1,L=0;len<ma+1;len<<=1,L++);
46     for(i=0;i<len;i++) if(s[i]<0) s[i]+=p;
47     FWT_XOR(s,len,1);
48     for(i=0;i<len;i++) 
49     {
50         if(s[i]>n) s[i]-=p; x=(n+s[i])/2;
51         s[i]=( ( ((n-x)&1) ? -1ll:1ll )*qpow(3,x)+p)%p;
52     }
53     FWT_XOR(s,len,-1);
54     printf("%d\n",(s[0]-1+p)%p);
55     return 0;
56 }

 

UOJ #310 黎明前的巧克力 (FWT)

标签:string   异或   其它   ace   函数   return   pac   算法理解   can   

原文地址:https://www.cnblogs.com/guapisolo/p/10597183.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!