欢迎访问 生活随笔!

生活随笔

当前位置: 首页 > 编程资源 > 编程问答 >内容正文

编程问答

2021牛客暑期多校训练营6 :D Gambling Monster 期望dp + fwt + cdq分治

发布时间:2023/12/4 编程问答 61 豆豆
生活随笔 收集整理的这篇文章主要介绍了 2021牛客暑期多校训练营6 :D Gambling Monster 期望dp + fwt + cdq分治 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

传送门

文章目录

  • 题意:
  • 思路:

题意:

给你一个大轮盘,被分成了nnn个区域0,1,2,..,n−10,1,2,..,n-10,1,2,..,n1,每个区域被转到的概率是ai∑j=0n−1aj\frac{a_i}{\sum_{j=0}^{n-1}a_j}j=0n1ajai,转到第iii个区域的时候得分是iii。现在你初始有x=0x=0x=0分,每次转动轮盘假设得到了yyy分,那么如果x⊕y≤xx\oplus y\le xxyx的话,当前得分不会变化,否则将x=x⊕yx=x\oplus yx=xy,当得分达到n−1n-1n1的时候立即停止,问你转转盘轮数的期望。
2≤n≤2162\le n\le2^{16}2n216,且nnn222的幂次。

思路:

以下我们约定:aia_iai代表得分为iii的概率。
考虑期望dpdpdp,我们这里倒着来推,所以设f[i]f[i]f[i]表示当得分为iii的时候还需要进行的轮数期望是多少,很容易就可以得到n2n^2n2的一个转移方程:fi=∑j=0n−1[(i⊕j)>i](fi⊕j+1)∗aj+∑j=0n−1[(i⊕j≤i)](fi+1)∗ajf_i=\sum_{j=0}^{n-1}[(i\oplus j)>i](f_{i\oplus j}+1)*a_j+\sum_{j=0}^{n-1}[(i\oplus j\le i)](f_i+1)*a_jfi=j=0n1[(ij)>i](fij+1)aj+j=0n1[(iji)](fi+1)aj
考虑到方程两边都有f[i]f[i]f[i],我们进行移项:
fi=∑j=0n−1aj+∑j=0n−1[(i⊕j)>i]fi⊕j∗aj1−∑j=0n−1[(i⊕j≤i)]ajf_i=\frac{\sum_{j=0}^{n-1}a_j+\sum_{j=0}^{n-1}[(i\oplus j)>i]f_{i\oplus j}*a_j}{1-\sum_{j=0}^{n-1}[(i\oplus j\le i)]a_j}fi=1j=0n1[(iji)]ajj=0n1aj+j=0n1[(ij)>i]fijaj
这个式子一眼看去只能n2n^2n2求,可以发现瓶颈就在∑j=0n−1[(i⊕j)>i]fi⊕j∗aj\sum_{j=0}^{n-1}[(i\oplus j)>i]f_{i\oplus j}*a_jj=0n1[(ij)>i]fijaj这个式子上,我们将其化简一下,设x=i,y=j,z=x⊕yx=i,y=j,z=x\oplus yx=i,y=j,z=xyf(x)=∑x⊕y=z,z>xf(z)a(y)f(x)=\sum_{x\oplus y=z,z>x}f(z)a(y)f(x)=xy=z,z>xf(z)a(y),看到f(x)f(x)f(x)的式子很像异或卷积,所以为了更直观,继续化简f(x)=∑z⊕y=x,z>xf(z)a(y)f(x)=\sum_{z\oplus y=x,z>x}f(z)a(y)f(x)=zy=x,z>xf(z)a(y)
观察可知,那么去掉z>xz>xz>x的条件的话,这个就是一个异或卷积的裸式子了,考虑到对于iii只有>i>i>ifff能对其有贡献,所以考虑cdqcdqcdq分治来计算这个式子,就可以去掉z>xz>xz>x这个条件了。在分治的过程中,只需要先递归计算右边的值,让后计算当前左区间的值,再递归左区间处理子问题即可。
但是问题还没有解决,a(y)a(y)a(y)怎么确定呢?在卷的过程中我们需要保证所有aaa都需要符合条件才可,不然会卷出来其他不合法的答案。考虑当前区间分成的两部分[l,mid],[mid+1,r][l,mid],[mid+1,r][l,mid],[mid+1,r]对应的二进制分别是[xxx000,xxx011],[xxx100,xxx111][xxx000,xxx011],[xxx100,xxx111][xxx000,xxx011],[xxx100,xxx111],发现了什么?我们计算左区间的时候,他们的aaa是那些部分呢?显然答案应该是aaa中下标在[100,111][100,111][100,111]的范围内的数,可以发现左区间[xxx000,xxx011][xxx000,xxx011][xxx000,xxx011]异或上区间内任何一个数都会准确的落在右区间的某个位置,且只有这些数能落在右区间。所以卷的时候让右区间的fff和上面的aaa对应区间卷起来即可。
递归到l=rl=rl=r的时候直接计算答案即可。

// Problem: Gambling Monster // Contest: NowCoder // URL: https://ac.nowcoder.com/acm/contest/11257/D // Memory Limit: 524288 MB // Time Limit: 2000 ms // // Powered by CP Editor (https://cpeditor.org)//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math") //#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native") //#pragma GCC optimize(2) #include<cstdio> #include<iostream> #include<string> #include<cstring> #include<map> #include<cmath> #include<cctype> #include<vector> #include<set> #include<queue> #include<algorithm> #include<sstream> #include<ctime> #include<cstdlib> #include<random> #include<cassert> #define X first #define Y second #define L (u<<1) #define R (u<<1|1) #define pb push_back #define mk make_pair #define Mid ((tr[u].l+tr[u].r)>>1) #define Len(u) (tr[u].r-tr[u].l+1) #define random(a,b) ((a)+rand()%((b)-(a)+1)) #define db puts("---") using namespace std;//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); } //void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); } //void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }typedef long long LL; typedef unsigned long long ULL; typedef pair<int,int> PII;const int N=100010,mod=1e9+7,INF=0x3f3f3f3f; const double eps=1e-6;int n; int a[N],b[N],c; LL f[N],iv2,u[N],all;LL qmi(LL a,LL b) {LL ans=1; a%=mod; a+=mod; a%=mod;while(b) {if(b&1) ans=ans*a%mod;a=a*a%mod;b>>=1;}return ans%mod; }int x[N], y[N]; void FWT_xor(int *a,int opt,int N) {for(int i=1;i<N;i<<=1)for(int p=i<<1,j=0;j<N;j+=p)for(int k=0;k<i;++k){int X=a[j+k],Y=a[i+j+k];a[j+k]=(X+Y)%mod;a[i+j+k]=(X+mod-Y)%mod;if(opt==-1)a[j+k]=1ll*a[j+k]*iv2%mod,a[i+j+k]=1ll*a[i+j+k]*iv2%mod;} }void cdq(int d,int l,int r,LL sa) {if(l==r) {f[l]+=all%mod; f[l]%=mod;(f[l]*=qmi(1-(all-sa)%mod,mod-2))%=mod;return;} int len=1<<d; LL sum=0;cdq(d-1,l+len,r,sa);for(int i=0;i<len;i++) {x[i]=a[i+len]; sum+=x[i]; sum%=mod;y[i]=f[i+l+len];}FWT_xor(x,1,len); FWT_xor(y,1,len);for(int i=0;i<len;i++) {x[i]=1ll*x[i]*y[i]%mod;}FWT_xor(x,-1,len);for(int i=0;i<len;i++) {(f[i+l]+=x[i])%=mod; }cdq(d-1,l,l+len-1,(sa+sum)%mod); }int main() { // ios::sync_with_stdio(false); // cin.tie(0);iv2=qmi(2,mod-2);int _; scanf("%d",&_);while(_--) {scanf("%d",&n); LL sum=0;memset(f,0,sizeof(f));for(int i=0;i<n;i++) scanf("%d",&a[i]),sum+=a[i];sum=qmi(sum,mod-2); all=0;for(int i=0;i<n;i++) a[i]=a[i]*sum%mod,all+=a[i],all%=mod;for(int i=16;i>=0;i--) {if((1<<i)==n) {cdq(i-1,0,n-1,0);break;}}printf("%lld\n",f[0]%mod);}return 0; } /* 200000005 3 3 0*/ 创作挑战赛新人创作奖励来咯,坚持创作打卡瓜分现金大奖

总结

以上是生活随笔为你收集整理的2021牛客暑期多校训练营6 :D Gambling Monster 期望dp + fwt + cdq分治的全部内容,希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错,欢迎将生活随笔推荐给好友。