【XSY2730】Ball 多项式exp 多项式ln 多项式开根 常系数线性递推 DP
题目大意
一行有\(n\)个球,现在将这些球分成\(k\) 组,每组可以有一个球或相邻两个球。一个球只能在至多一个组中(可以不在任何组中)。求对于\(1\leq k\leq m\)的所有\(k\)分别有多少种分组方法。
答案对\(998244353\)取模。
\(n\leq {10}^9,m<2^{19}\)
题解
因为\(k>n\)的项都是\(0\),所以我们钦定\(m\leq n\)
考虑DP。
记\(f_{i,j}\)为前\(i\)个球分为\(j\)组的方案数。
\]
直接做是\(O(nm)\)的。
如果把\(f_i\)看成一个多项式,即
\]
那么转移就变成了
\]
这是一个常系数齐次线性递推,用FFT优化可以做到\(O(m\log m\log n)\)
考虑怎么求一个常系数齐次线性递推关系的通项公式。
先求出这个转移矩阵的特征多项式:
\]
特征值为
\lambda_1&=\frac{1+x+\sqrt{x^2+6x+1}}{2}\\
\lambda_2&=\frac{1+x-\sqrt{x^2+6x+1}}{2}
\end{align}
\]
我们钦定\(F_{-1}(x)=0\),设
\]
带入\(F_{-1}(x),F_0(x)\)得
c_1&+c2&=0\\
c_2\lambda_1&+c_2\lambda_2&=1
\end{cases}
\]
解得
c_1&=\frac{1}{\lambda_1-\lambda_2}\\
c_2&=\frac{1}{\lambda_2-\lambda_1}
\end{cases}
\]
于是
\]
直接用多项式开根求出\(\lambda_1,\lambda_2\),然后用多项式\(\ln \exp\)求出\(F_n(x)\)
时间复杂度:\(O(m\log m)\)
小优化:因为\([x^0]\lambda_2=0\),所以\(\lambda_2\)的前\(n+1\)项都是\(0\)。因为\(m\leq n\),所以
\]
所以我们不用计算\({\lambda_2}^{n+1}\)了。
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=300000;
const ll p=998244353;
const ll g=3;
const ll inv2=(p+1)/2;
ll fp(ll a,ll b)
{
ll s=1;
for(;b;b>>=1,a=a*a%p)
if(b&1)
s=s*a%p;
return s;
}
int rev[maxn];
ll w1[maxn];
ll w2[maxn];
void ntt(ll *a,int n,int t)
{
int i,j,k;
ll u,v,w,wn;
for(i=1;i<n;i++)
{
rev[i]=(rev[i>>1]>>1)|(i&1?n>>1:0);
if(rev[i]>i)
swap(a[i],a[rev[i]]);
}
for(i=2;i<=n;i<<=1)
{
wn=(t==1?w1[i]:w2[i]);
for(j=0;j<n;j+=i)
{
w=1;
for(k=j;k<j+i/2;k++)
{
u=a[k];
v=a[k+i/2]*w%p;
a[k]=(u+v)%p;
a[k+i/2]=(u-v)%p;
w=w*wn%p;
}
}
}
if(t==-1)
{
ll inv=fp(n,p-2);
for(i=0;i<n;i++)
a[i]=a[i]*inv%p;
}
}
void getinv(ll *a,ll *b,int n)
{
if(n==1)
{
b[0]=fp(a[0],p-2);
return;
}
getinv(a,b,n>>1);
static ll a1[maxn],a2[maxn];
int i;
for(i=0;i<n;i++)
a1[i]=a[i];
for(;i<n<<1;i++)
a1[i]=0;
for(i=0;i<n>>1;i++)
a2[i]=b[i];
for(;i<n<<1;i++)
a2[i]=0;
ntt(a1,n<<1,1);
ntt(a2,n<<1,1);
for(i=0;i<n<<1;i++)
a1[i]=a2[i]*((2-a1[i]*a2[i])%p)%p;
ntt(a1,n<<1,-1);
for(i=0;i<n;i++)
b[i]=a1[i];
}
void getsqrt(ll *a,ll *b,int n)
{
if(n==1)
{
b[0]=1;
return;
}
getsqrt(a,b,n>>1);
static ll a1[maxn],a2[maxn],a3[maxn];
int i;
for(i=0;i<n>>1;i++)
a1[i]=b[i];
for(;i<n<<1;i++)
a1[i]=0;
for(i=0;i<n;i++)
a2[i]=a[i];
for(;i<n<<1;i++)
a2[i]=0;
getinv(a1,a3,n);
ntt(a2,n<<1,1);
ntt(a3,n<<1,1);
for(i=0;i<n<<1;i++)
a2[i]=a2[i]*a3[i]%p;
ntt(a2,n<<1,-1);
for(i=0;i<n;i++)
b[i]=(a1[i]+a2[i])%p*inv2%p;
}
void getln(ll *a,ll *b,int n)
{
static ll inv[maxn],a1[maxn],a2[maxn];
int i;
inv[0]=inv[1]=1;
for(i=2;i<n;i++)
inv[i]=-p/i*inv[p%i]%p;
for(i=1;i<n;i++)
a1[i-1]=a[i]*i%p;
a1[n-1]=0;
getinv(a,a2,n);
for(i=n;i<n<<1;i++)
a1[i]=a2[i]=0;
ntt(a1,n<<1,1);
ntt(a2,n<<1,1);
for(i=0;i<n<<1;i++)
a1[i]=a1[i]*a2[i]%p;
ntt(a1,n<<1,-1);
for(i=1;i<n;i++)
b[i]=a1[i-1]*inv[i]%p;
b[0]=0;
}
void getexp(ll *a,ll *b,int n)
{
if(n==1)
{
b[0]=1;
return;
}
getexp(a,b,n>>1);
int i;
for(i=n>>1;i<n;i++)
b[i]=0;
static ll a1[maxn],a2[maxn],a3[maxn];
getln(b,a1,n);
for(i=0;i<n>>1;i++)
{
a2[i]=b[i];
a3[i]=a[i+(n>>1)]-a1[i+(n>>1)];
}
for(;i<n;i++)
a2[i]=a3[i]=0;
ntt(a2,n,1);
ntt(a3,n,1);
for(i=0;i<n;i++)
a2[i]=a2[i]*a3[i]%p;
ntt(a2,n,-1);
for(i=0;i<n>>1;i++)
b[i+(n>>1)]=a2[i];
}
ll a[maxn];
ll b[maxn];
ll c[maxn];
ll d[maxn];
ll e[maxn];
int n,m;
int k;
void solve()
{
k=1;
while(k<=m)
k<<=1;
int i;
for(i=2;i<=k<<1;i++)
{
w1[i]=fp(g,(p-1)/i);
w2[i]=fp(w1[i],p-2);
}
d[0]=1;
d[1]=6;
d[2]=1;
getsqrt(d,c,k);
for(i=0;i<k;i++)
a[i]=c[i];
a[0]++;
a[1]++;
for(i=0;i<k;i++)
a[i]=a[i]*inv2%p;
getinv(c,b,k);
getln(a,e,k);
for(i=0;i<k;i++)
e[i]=e[i]*(n+1)%p;
getexp(e,a,k);
for(i=k;i<k<<1;i++)
a[i]=b[i]=0;
ntt(a,k<<1,1);
ntt(b,k<<1,1);
for(i=0;i<k<<1;i++)
a[i]=a[i]*b[i]%p;
ntt(a,k<<1,-1);
for(i=1;i<=m;i++)
{
a[i]=(a[i]+p)%p;
printf("%lld ",a[i]);
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("b.in","r",stdin);
freopen("b.out","w",stdout);
#endif
int x=0;
scanf("%d%d",&n,&m);
if(m>n)
{
x=m-n;
m=n;
}
solve();
while(x--)
printf("0 ");
return 0;
}
【XSY2730】Ball 多项式exp 多项式ln 多项式开根 常系数线性递推 DP的更多相关文章
- 【BZOJ4944】【NOI2017】泳池 概率DP 常系数线性递推 特征多项式 多项式取模
题目大意 有一个\(1001\times n\)的的网格,每个格子有\(q\)的概率是安全的,\(1-q\)的概率是危险的. 定义一个矩形是合法的当且仅当: 这个矩形中每个格子都是安全的 必须紧贴网格 ...
- Solution -「LOJ #150」挑战多项式 ||「模板」多项式全家桶
\(\mathcal{Description}\) Link. 给定 \(n\) 次多项式 \(F(x)\),在模 \(998244353\) 意义下求 \[G(x)\equiv\left\{ ...
- [模板]多项式全家桶小记(求逆,开根,ln,exp)
前言 这里的全家桶目前只包括了\(ln,exp,sqrt\).还有一些类似于带余数模,快速幂之类用的比较少的有时间再更,\(NTT\)这种前置知识这里不多说. 还有一些基本的导数和微积分内容要了解,建 ...
- 多项式的各类计算(多项式的逆/开根/对数/exp/带余除法/多点求值)
预备知识:FFT/NTT 多项式的逆 给定一个多项式 F(x)F(x)F(x),请求出一个多项式 G(x)G(x)G(x),满足 F(x)∗G(x)≡1(mod xn)F(x)*G(x) \equiv ...
- 牛顿迭代,多项式求逆,除法,开方,exp,ln,求幂
牛顿迭代 若 \[G(F_0(x))\equiv 0(mod\ x^{2^t})\] 牛顿迭代 \[F(x)\equiv F_0(x)-\frac{G(F_0(x))}{G'(F_0(x))}(mod ...
- 【xsy2978】Product of Roots 生成函数+多项式ln+多项式exp
题目大意:给你两个多项式$f(x)$和$g(x)$,满足$f(x)=\prod\limits_{i=1}^{n}(a_i+1)$,$g(x)=\prod\limits_{i=1}^{m}(b_i+1) ...
- 洛谷P5245 【模板】多项式快速幂(多项式ln 多项式exp)
题意 题目链接 Sol \(B(x) = \exp(K\ln(A(x)))\) 做完了... 复杂度\(O(n\log n)\) // luogu-judger-enable-o2 // luogu- ...
- 【WC2019】数树 树形DP 多项式exp
题目大意 有两棵 \(n\) 个点的树 \(T_1\) 和 \(T_2\). 你要给每个点一个权值吗,要求每个点的权值为 \([1,y]\) 内的整数. 对于一条同时出现在两棵树上的边,这条边的两个端 ...
- 洛谷P4726 【模板】多项式指数函数(多项式exp)
题意 题目链接 Sol 多项式exp,直接套泰勒展开的公式 \(F(x) = e^{A(x)}\) 求个导\(F'(x) = A(x)\) 我们要求的就是\(G(f(x)) = lnF(x) - A( ...
随机推荐
- python中各个response使用
Python django中我们经常用的response有django中的 JsonResponse, HttpResponse,还有DRF中的Response 在使用的时候,经常会不知道如何什么时候 ...
- Graph Without Long Directed Paths CodeForces - 1144F (dfs染色)
You are given a connected undirected graph consisting of nn vertices and mm edges. There are no self ...
- Array Queries CodeForces - 797E
题目链接 非常好的一道题目, 分析,如果用暴力的话,时间复杂度是O(q*n)稳稳的超时 如果用二维DP的话,需要O (n*n)的空间复杂度,会爆空间. 那么分析一下,如果k>sqrt(n)的话, ...
- Elasticsearch 关键字:索引,类型,字段,索引状态,mapping,文档
1. 索引(_index)索引:说的就是数据库的名字.我这个说法是对应到咱经常使用的数据库. 结合es的插件 head 来看. 可以看到,我这个地方,就有这么几个索引,索引就是数据库,后面是这个数据库 ...
- 学习yii2.0——行为
学习yii框架的行为之前,需要先了解yii的事件,可以参考这篇博客: 怎么理解行为 yii框架的行为有点类似于trait,可以有个大体的概念: 1.有一个类A,包含一些属性和方法,很普通的一个类A. ...
- PAT L3-016 二叉搜索树的结构
https://pintia.cn/problem-sets/994805046380707840/problems/994805047903240192 二叉搜索树或者是一棵空树,或者是具有下列性质 ...
- noode inquirer
一. 由于交互的问题种类不同,inquirer为每个问题提供很多参数: type:表示提问的类型,包括:input, confirm, list, rawlist, expand, checkbox, ...
- [转帖]Windows NT 之父 - David Cutler
Windows NT 之父 - David Cutler https://www.cnblogs.com/wangwust/p/6826200.html 曾经下过 夺路狂奔的电子书 但是还没看完.. ...
- input type=date时,时间数据回填,报错The specified value "2019-0404-18" does not conform to the required format, "yyyy-MM-dd".
<input autocomplete id="start-time" name="start_time" type="date" c ...
- docker 列出每个容器的IP
抄来的...找不到出处了. 常用方法有两种 docker inspect 容器ID | grep IPAddress 方法二 查看docker name: sudo docker inspect ...