乒乓球

题目描述

小 BoBoBo 是某省乒乓球名列前茅的选手，现在他有 nnn 颗乒乓球一字排开，第iii颗乒乓球的权值为 wiw_iwi
每次他会随机从现有的乒乓球中等概率选一颗拿走，然后得到的收益是这颗球左边第一个乒乓球和右边第一个乒乓球的权值的乘积，如果左边没有乒乓球或者右边没有乒乓球，则收益为 000，这个过程会重复进行到所有球都被拿走为止
现在小 BoBoBo 想知道他的期望总收益
为了方便，你只需要输出答案对 998244353998244353998244353 取模的值.

解决方案

思路就是枚举一对乒乓球(i,j)(i,j)(i,j)然后计算它对于答案的贡献.
(i,j)(i,j)(i,j)要想产生贡献,那么至少要满足2≤j−i≤n−12 \le j-i \le n-12≤j−i≤n−1.

考虑i,ji,ji,j两颗乒乓球一定要在[i+1,j−1][i+1,j-1][i+1,j−1]之间的乒乓球全都取完才能取,这样的情况共有(j−i−1)!∗2!∗Cnj−i+1∗(n−(j−i+1))!(j-i-1)!*2!*C_n^{j-i+1}*(n-(j-i+1))!(j−i−1)!∗2!∗Cnj−i+1∗(n−(j−i+1))!

也就是2∗n!(j−i)(j−i+1)\frac{2*n!}{(j-i)(j-i+1)}(j−i)(j−i+1)2∗n!,而最后的答案要除以n!n!n!,所以在这里直接除掉就可以了,也就是2(j−i)(j−i+1)\frac{2}{(j-i)(j-i+1)}(j−i)(j−i+1)2

共线即为wi∗wj∗2(j−i)(j−i+1)w_i*w_j*\frac{2}{(j-i)(j-i+1)}wi∗wj∗(j−i)(j−i+1)2

而这个值只与(j−i)(j-i)(j−i)的大小有关,而如果我们将式子中第二个wjw_jwj反转,即改写成bn−jb_{n-j}bn−j.(其中bi=wn−ib_i = w_{n-i}bi=wn−i)

那么wi∗bn−j∗2(j−i)(j−i+1)w_i*b_{n-j}*\frac{2}{(j-i)(j-i+1)}wi∗bn−j∗(j−i)(j−i+1)2

这下我们枚举2(t)(t+1)\frac{2}{(t)(t+1)}(t)(t+1)2,找所有满足的(i,n−j)(i,n-j)(i,n−j)对,使得n−j−i=n−tn-j-i = n-tn−j−i=n−t

显然这就是一个求卷积的裸题了,www序列与bbb序列卷积的第n−tn-tn−t项就是我们要求的答案.

代码

#include <iostream>
#include <cstring>
#include <cstdio>using namespace std;
typedef long long LL;
const int N = 1 << 20;
const int P = 998244353;
const int G = 3;
const int NUM = 20;LL  wn[NUM];
LL  a[N], b[N];LL quick_mod(LL a, LL b, LL m)
{LL ans = 1;a %= m;while(b){if(b & 1){ans = ans * a % m;b--;}b >>= 1;a = a * a % m;}return ans;
}void GetWn()
{for(int i = 0; i < NUM; i++){int t = 1 << i;wn[i] = quick_mod(G, (P - 1) / t, P);}
}
void Rader(LL a[], int len)
{int j = len >> 1;for(int i = 1; i < len - 1; i++){if(i < j) swap(a[i], a[j]);int k = len >> 1;while(j >= k){j -= k;k >>= 1;}if(j < k) j += k;}
}void NTT(LL a[], int len, int on)
{Rader(a, len);int id = 0;for(int h = 2; h <= len; h <<= 1){id++;for(int j = 0; j < len; j += h){LL w = 1;for(int k = j; k < j + h / 2; k++){LL u = a[k] % P;LL t = w * a[k + h / 2] % P;a[k] = (u + t) % P;a[k + h / 2] = (u - t + P) % P;w = w * wn[id] % P;}}}if(on == -1){for(int i = 1; i < len / 2; i++)swap(a[i], a[len - i]);LL inv = quick_mod(len, P - 2, P);for(int i = 0; i < len; i++)a[i] = a[i] * inv % P;}
}void Conv(LL a[], LL b[], int n)
{NTT(a, n, 1);NTT(b, n, 1);for(int i = 0; i < n; i++)a[i] = a[i] * b[i] % P;NTT(a, n, -1);
}
LL Fac[N];
int n;
int main()
{   Fac[0] = 1;for(int i = 1;i < N;++i) {Fac[i] = Fac[i-1] * i % P;}GetWn();std::cin >> n;for(int i = 0;i < n;++i){ std::cin >> a[i];b[n-i] = a[i];}int len = 1;while(len < n) len <<= 1;len <<= 1;Conv(a,b,len);LL ans = 0;for(int le = 2;le <= n-1;++le) {LL part = 2 * quick_mod(le,P-2,P) % P * quick_mod(le+1,P-2,P) % P;ans = (ans + (a[n-le]*part % P)) % P;}std::cout << ans << std::endl;return 0;
}

数学推导题,NTT,快速数论变换,Wannafly-乒乓球相关推荐

数学推导题,NTT,快速数论变换,Wannafly-导数卷积
导数卷积题目描述题解参考了一下标程的推导过程,因为这个推导对我这种数学弱渣真的有点难鸭. [1]f(x)f(x)f(x)的iii次导函数: f(i)(x)=ai∗i!0!+ai+1∗(i+1)! ...
FFT 快速傅里叶变换 NTT 快速数论变换
SDNU 1531 a*b III (FFT模板) Description 计算a乘b,多组输入(50组以内). Input 输入a b,数据范围0 <= a,b <= 10^100000 ...
A*B NTT快速数论变换
wmq的A×B Problem 发布时间: 2017年4月9日 17:06 最后更新: 2017年4月9日 17:07 时间限制: 3000ms 内存限制: 512M 描述这是一个非常简 ...
NTT（快速数论变换）模板
NTT好文 NTT快速数论变换,在acm中用于解决一类多项式卷积问题. 多项式A和多项式B求卷积后的多项式为C. 显然,如果A是n次多项式,B有m次多项式,这两个多项式都不缺项的话,乘积应为n+m次多 ...
【学习笔记】超简单的快速数论变换（NTT）（FFT的优化）（含全套证明）
整理的算法模板合集: ACM模板目录一.前置知识二.快速数论变换(NTT) 三.NTT证明(和FFT的关系) 四.NTT模板数组形式的实现 vector形式的实现点我看多项式全家桶(●^◡_ ...
多项式算法2：NTT（快速数论变换）
多项式算法2:NTT(快速数论变换) 前言前置知识正文前言算法简介上次的FFT大量使用浮点数,有精度不好控制的问题,使用可以保证精度的方法又容易超时.这里NTT是在取模的一个前提下找到一个新 ...
比FFT还容易明白的NTT（快速数论变换）
NTT相关一种快速数论变换算法,这种算法是以数论为基础,对样本点为的数论变换,按时间抽取的方法,得到一组等价的迭代方程,有效高速简化了方程中的计算公式·与直接计算相比,大大减少了运算次数.(见快速傅 ...
算法学习FFT系列(2):快速数论变换NTT bzoj3992: [SDOI2015]序列统计例题详解
bzoj3992: [SDOI2015]序列统计 Description 小C有一个集合S,里面的元素都是小于M的非负整数.他用程序编写了一个数列生成器,可以生成一个长度为N的数列,数列中的每个数都属 ...
python蝴蝶代码_快速数论变换（NTT）及蝴蝶操作构造详解
快速数论变换本文不会从头开始介绍NTT算法,所以需要先了解FFT:永远在你身后:快速傅里叶变换(FFT)求解多项式乘法zhuanlan.zhihu.com 除此之外,先简单的铺垫一些数论的概念同余 ...

数学推导题,NTT,快速数论变换,Wannafly-乒乓球

乒乓球

题目描述

解决方案

代码

数学推导题,NTT,快速数论变换,Wannafly-乒乓球相关推荐

最新文章

热门文章