BZOJ 4555 [Tjoi2016&Heoi2016]求和 (多项式求逆)
题目链接:
https://www.lydsy.com/JudgeOnline/problem.php?id=4555
题目大意:
给定 \(S(n,m)\) 表示第二类斯特林数,定义函数 \(f(n)\) :
\(f(n) = \sum_{i=0}^n\sum_{j=0}^iS(i,j)*2^j*(j!)\)
\(S(i, j)\) 表示第二类斯特林数,递推公式为:
\(S(i,j) = S(i-1,j-1) + j*S(i-1,j),(1 \leq j \leq i-1)\).
边界条件为:\(S(i, i) = 1(0 <= i), S(i, 0) = 0(1 <= i)\).
给定正整数 \(n\) ,\((n≤10^5)\),求\(f(n)\)?
前置知识:
考虑将 \(n\) 个不同的球放入\(m\)个不同的盒子内,但没有无空盒限制(可以有空盒)的方案数 就是\(m^n\)。但是不能有空盒就是第二类斯特林数。
第二类斯特林数\(S(n,m)\)的意义是把 \(n\) 个元素划分成 \(m\) 个无序的集合的方案,就是把\(n\)个有特征(不相同)的球放入\(m\)个没有特征(相同)的盒子的方案数(每一个盒子里面都至少有一个球)的方案数。
题解:
设\(g(n) = \sum_{i=0}^nS(n,i)*2^i*(i!)\),那么我们要求的是这个式子的前缀和。
考虑一下 \(g(n)\)的意义,\(g(n)\)的意义就是把\(1\)~\(n\)中不同的数放入若干个不同的集合中,且每一个集合有两种状态的方案数。
所以可以把式子写出这样(注意: 下标从 \(1\) 开始):
\(g(n)=\sum\limits_{i=1}^n 2*{n \choose i}*g(n-i)\),意思是枚举第一个集合的方案数再乘以剩余集合的方案数。
将上式写出卷积的形式:
\(g(n)=n! * \sum\limits_{i=1}^n \frac{2}{i!}*\frac{g(n-i)}{(n-i)!}\).
\(\frac{g(n)}{n!}=\sum\limits_{i=1}^n \frac{2}{i!}*\frac{g(n-i)}{(n-i)!}\).
观察一下右表达式。
这不就是指数型生成函数吗。
可以令:(注意:这两个式子的下标不一样。)
\(F(x)=\sum_{i=0}^\infty {g(i)\over i! } x^i\)
\(G(x)=\sum_{i=1}^\infty {2\over i!}x^i\)。
那么有下面这个式子:
\(G(x)=F(x) * G(x) +1\).
注意:这里的 \(1\) 实际上指的是,由于\(G(x)\)和\(F(x)\)的起始下标不同,选择以 \(1\) 为下标开始卷积,所以需要舍弃掉 \(\frac{g(n-0)}{(n-0)!}\) 这一项,再补上 \(F(0)\) 的系数。
最后有:
\(G(x)=\frac{1}{1-F(x)}\)。
这个式子显然可以多项式求逆。
然后构造出\(F(x)\)再通过多项式求逆,就可以得到 \(G(x)\) 的系数,求和可以得到最终答案。
复杂度:\(O(nlogn)\)。
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 10;
const int maxLen = 18, maxm = 1 << maxLen | 1;
const ll maxv = 1e10 + 6; // 1e14, 1e15
const long double pi = acos(-1.0); // double maybe is not enough
ll mod = 998244353, nlim, sp, msk;
// https://www.lydsy.com/JudgeOnline/problem.php?id=4555
ll qpower(ll x, ll p) { // x ^ p % mod
ll ret = 1;
while (p) {
if (p & 1) (ret *= x) %=mod;
(x *= x) %=mod;
p >>= 1;
}
return ret;
}
struct cp {
long double r, i;
cp() {}
cp(long double r, long double i) : r(r), i(i) {}
cp operator + (cp const &t) const { return cp(r + t.r, i + t.i); }
cp operator - (cp const &t) const { return cp(r - t.r, i - t.i); }
cp operator * (cp const &t) const { return cp(r * t.r - i * t.i, r * t.i + i * t.r); }
cp conj() const { return cp(r, -i); }
} w[maxm], wInv[maxm];
void init() {
for(int i = 0, ilim = 1 << maxLen; i < ilim; ++i) {
int j = i, k = ilim >> 1; // 2 pi / ilim
for( ; !(j & 1) && !(k & 1); j >>= 1, k >>= 1);
w[i] = cp((long double)cos(pi / k * j), (long double)sin(pi / k * j));
wInv[i] = w[i].conj();
}
nlim = std::min(maxv / (mod - 1) / (mod - 1), maxn - 1LL);
for(sp = 1; 1 << (sp << 1) < mod; ++sp);
msk = (1 << sp) - 1;
}
void FFT(int n, cp a[], int flag) {
static int bitLen = 0, bitRev[maxm] = {};
if(n != (1 << bitLen)) {
for(bitLen = 0; 1 << bitLen < n; ++bitLen);
for(int i = 1; i < n; ++i)
bitRev[i] = (bitRev[i >> 1] >> 1) | ((i & 1) << (bitLen - 1));
}
for(int i = 0; i < n; i ++) {
if(i < bitRev[i]) {
std::swap(a[i], a[bitRev[i]]);
}
}
for(int i = 1, d = 1; d < n; ++i, d <<= 1)
for(int j = 0; j < n; j += d << 1)
for(int k = 0; k < d; ++k) {
cp &AL = a[j + k], &AH = a[j + k + d];
cp TP = w[k << (maxLen - i)] * AH;
AH = AL - TP, AL = AL + TP;
}
if(flag != -1)
return;
std::reverse(a + 1, a + n);
for(int i = 0; i < n; ++i) {
a[i].r /= n;
a[i].i /= n;
}
}
void polyMul(int a[], int aLen, int b[], int bLen, int c[]) // a 和 b 的卷积 c
{
// std::cout << "mod = " << mod << '\n';
static cp A[maxm], B[maxm], C[maxm], D[maxm];
int len, cLen = aLen + bLen - 1;
for(len = 1; len < aLen + bLen - 1; len <<= 1);
if(std::min(aLen, bLen) <= nlim)
{
for(int i = 0; i < len; i++) {
A[i] = cp(i < aLen ? a[i] : 0, i < bLen ? b[i] : 0);
}
FFT(len, A, 1);
cp tr(0, -0.25);
for(int i = 0, j; i < len; i++) {
j = (len - i) & (len - 1), B[i] = (A[i] * A[i] - (A[j] * A[j]).conj()) * tr;
}
FFT(len, B, -1);
for(int i = 0; i < cLen; ++i) c[i] = (ll)(B[i].r + 0.5) % mod;
return;
} // if min(aLen, bLen) * mod <= maxv
for(int i = 0; i < len; ++i) {
A[i] = i < aLen ? cp(a[i] & msk, a[i] >> sp) : cp(0.0, 0.0);
B[i] = i < bLen ? cp(b[i] & msk, b[i] >> sp) : cp(0.0, 0.0);
}
FFT(len, A, 1);
FFT(len, B, 1);
cp trL(0.5, 0.0), trH(0.0, -0.5), tr(0.0, 1.0);
for(int i = 0, j; i < len; i++) {
j = (len - i) & (len - 1);
cp AL = (A[i] + A[j].conj()) * trL;
cp AH = (A[i] - A[j].conj()) * trH;
cp BL = (B[i] + B[j].conj()) * trL;
cp BH = (B[i] - B[j].conj()) * trH;
C[i] = AL * (BL + BH * tr);
D[i] = AH * (BL + BH * tr);
}
FFT(len, C, -1);
FFT(len, D, -1);
for(int i = 0; i < cLen; ++i)
{
int v11 = (ll)(C[i].r + 0.5) % mod, v12 = (ll)(C[i].i + 0.5) % mod;
int v21 = (ll)(D[i].r + 0.5) % mod, v22 = (ll)(D[i].i + 0.5) % mod;
c[i] = (((((ll)v22 << sp) + v12 + v21) << sp) + v11) % mod;
}
}
int c[maxm], tmp[maxm];
void polyInv(int x[], int y[], int deg) { // 对多项式 x 求逆
if (deg == 1) {
y[0] = qpower(x[0], mod - 2);
return;
}
polyInv(x, y, (deg + 1) >> 1);
copy(x, x + deg, tmp);
int p = ((deg + 1) >> 1) + deg - 1;
polyMul(y, (deg + 1) >> 1, tmp, deg, c);
for (int i = 0; i < p; i += 1) c[i] = (- c[i] + mod) %mod;
(c[0] += 2) %=mod;
polyMul(y, (deg + 1) >> 1, c, deg, tmp);
copy(tmp, tmp + deg, y);
}
int F[maxn],G[maxn];
ll inv[maxn];
ll fac[maxn];
ll ans;
ll n,m;
int main()
{
init();
std::cin >> n;
m = 1;
while(m <= n) m<<=1;
fac[0]=1;
for(ll i = 1;i <= m;i++) {
fac[i] = fac[i-1] * i % mod;
}
inv[0] = inv[1] = 1;
inv[m] = qpower(fac[m],mod - 2);
for(ll i = m;i >= 1;--i) {
inv[i - 1] = inv[i] * i % mod;
}
F[0] = 1;
for(int i = 1;i <= n;i++) {
F[i] = (-inv[i] + mod) * 2 % mod;
}
polyInv(F,G,m); // 得到多项式 G(x)的系数
ans = 0;
for(int i = n;i >= 0;--i) { // 求和得到答案
ans += G[i] * fac[i] % mod;
ans %= mod;
}
std::cout << ans << '\n';
return 0;
}
BZOJ 4555 [Tjoi2016&Heoi2016]求和 (多项式求逆)的更多相关文章
- 【BZOJ 4555】[Tjoi2016&Heoi2016]求和 多项式求逆/NTT+第二类斯特林数
出处0.0用到第二类斯特林数的性质,做法好像很多,我打的是直接ntt,由第二类斯特林数的容斥公式可以推出,我们可以对于每一个i,来一次ntt求出他与所有j组成的第二类斯特林数的值,这个时候我们是O(n ...
- BZOJ 4555: [Tjoi2016&Heoi2016]求和 [分治FFT 组合计数 | 多项式求逆]
4555: [Tjoi2016&Heoi2016]求和 题意:求\[ \sum_{i=0}^n \sum_{j=0}^i S(i,j)\cdot 2^j\cdot j! \\ S是第二类斯特林 ...
- BZOJ 4555: [Tjoi2016&Heoi2016]求和 [FFT 组合计数 容斥原理]
4555: [Tjoi2016&Heoi2016]求和 题意:求\[ \sum_{i=0}^n \sum_{j=0}^i S(i,j)\cdot 2^j\cdot j! \\ S是第二类斯特林 ...
- BZOJ 4555 [Tjoi2016&Heoi2016]求和 ——分治 NTT 多项式求逆
不想多说了,看网上的题解吧,我大概说下思路. 首先考察Stirling的意义,然后求出递推式,变成卷积的形式. 然后发现贡献是一定的,我们可以分治+NTT. 也可以直接求逆(我不会啊啊啊啊啊) #in ...
- bzoj 4555 [Tjoi2016&Heoi2016]求和 NTT 第二类斯特林数 等比数列求和优化
[Tjoi2016&Heoi2016]求和 Time Limit: 40 Sec Memory Limit: 128 MBSubmit: 679 Solved: 534[Submit][S ...
- bzoj 4555: [Tjoi2016&Heoi2016]求和【NTT】
暴力推式子推诚卷积形式,但是看好多blog说多项式求逆不知道是啥.. \[ \sum_{i=0}^{n}\sum_{j=0}^{n}S(i,j)*2^j*j! \] \[ S(i,j)=\frac{1 ...
- [BZOJ 4555][Tjoi2016&Heoi2016]求和
题意 给定 $n$ , 求下式的值: $$ f(n)= \sum_{i=0}^n\sum_{j=0}^i\begin{Bmatrix}i\\ j\end{Bmatrix}\times 2^j\time ...
- BZOJ 4555: [Tjoi2016&Heoi2016]求和 (NTT + 第二类斯特林数)
题意 给你一个数 \(n\) 求这样一个函数的值 : \[\displaystyle f(n)=\sum_{i=0}^{n}\sum_{j=0}^{i} \begin{Bmatrix} i \\ j ...
- BZOJ 4555:[TJOI2016&HEOI2016]求和(第二类斯特林数+NTT)
题目链接 \(Description\) 求 \[\sum_{i=0}^n\sum_{j=0}^iS(i,j)2^jj!\]对998244353取模后的结果. \(n<=10^5\) \(Sol ...
随机推荐
- 在performancepoint里面建立数据源的时候,总是发生以下的报警
就是我在performancepoint里面建立数据源的时候,总是发生以下的报警. 在管理中心主页的“应用程序管理”部分,单击“管理服务应用程序”,然后单击“PerformancePoint Se ...
- 关于【搭建LAMP环境时,php测试页面打不开】解决
关于[搭建LAMP环境时,php测试页面打不开]解决 〇.我的测试页面是: http://172.30.124.10/index.php 用火狐打不开,如下图. 一.httpd已经启动了(system ...
- 【Henu ACM Round#17 A】Simple Game
[链接] 我是链接,点我呀:) [题意] 在这里输入题意 [题解] 肯定是放在m-1或者m+1的. (m-1是左边的点都离a最近,而m+1则是右边的点都离他最近. 看看哪个更好就行 [代码] #inc ...
- IT运维分析
http://www.infoq.com/cn/presentations/case-of-it-operation-analysis-and-log-search-analysis-engine?u ...
- Android Studio升级到0.8.1后怎样设置字体大小?
升级到0.8.1后.打开设置字体大小页面.你会发现无论是Default还是Darcula,都不同意你改变字体的大小.事实上这个是由于这两个模式是Android Studio自带模式,所以不同意你修改, ...
- 创建maven项目pom.xml出现错误(依赖Missing)
Maven的依赖问题 在聚合模块时候,发现在父工程目录中的依赖存在一些问题.一开始是${pagehelper.version},后来将版本直接填写相应的版本如图下 依赖添加失败 * 在父工程的jar包 ...
- 下载eclipse详细步骤
先登陆eclipse的官网 然后点击红色箭头进行选择你电脑是32还是64位的 根据自己的需求下载 然后点击下载 这里下载的是安装包,你要进行压缩.安装时一定要好相应的jdk要不然就会报错 这上面的错误 ...
- 5.cocos2d锚点
创建一个层T1LayerAnchorPoint AppDelegate.cpp bool AppDelegate::applicationDidFinishLaunching() { // initi ...
- Kinect 开发 —— 进阶指引 (下)
运动识别 利用运动识别(motion detection)来进行近景识别是最有意思的一种方式.实现运动识别的基本原理是设置一个起始的基准RGB图像,然后将从摄像头获取的每一帧影像和这个基准图像进行比较 ...
- 使用Java语言开发微信公众平台(三)
在上一节课程中,我们来学习了微信公众平台最基础的一个接口——access_token,并且能够从微信公众平台中取到access_token. 那么,在本节课程中,我们要以上节课获取到的 ...