【洛谷5437】【XR-2】约定(拉格朗日插值)

题面

洛谷

题解

首先发现每条边除了边权之外都是等价的,所以可以考虑每一条边的出现次数。

显然钦定一条边之后构成生成树的方案数是\(2*n^{n-3}\)。可以直接\(purfer\)序列算。

也可以发现每一条边的出现次数相等,树的总数是\(n^{n-2}\),每次出现\(n-1\)条边,每条边又是等价的。

也可以算出上面这个值。

于是要算的东西就变成了

\[\displaystyle \sum_{i=1}^n\sum_{j=i+1}^n(i+j)^k
\]

这个东西不对称,很不方便计算,所以可以变成:

\[\frac{1}{2}(\sum_{i=1}^n \sum_{j=1}^n (i+j)^k-\sum_{i=1}^n (i+i)^k)
\]

\(k\)次方这个东西显然是个\(k+1\)次多项式,可以套进去直接拉格朗日插值计算。

拆一下变成了\(\displaystyle \sum_{i=1}^{n} (i-1) i^k+\sum_{i=n+1}^{2n}(2n-i+1)i^k-\sum_{i=1}^n 2^ki^k\)。

然后预处理之后,可以用拉格朗日插值可以在\(O(k)\)的复杂度里算出上面的式子,然后带回去算期望就行了。

然后这里怎么拉格朗日插值。

以第一个函数为例。

令\(f(n)=\sum_{i=1}^n (i-1)i^k\),因为\(i^{k+1}\)次方大概是一个\(k+2\)次多项式,所以我们需要\(k+3\)个值,那么显然这个函数的前\(k+3\)项我们在预处理之后是可以提前算出来的。

然后根据拉格朗日插值的公式,对于一个\(k\)次多项式而言:

\[P(x)=\sum_{i=1}^{k+1}P(x_i)\prod_{j=1,j\neq i}^{k+1}\frac{x-x_j}{x_i-x_j}
\]

然后因为我们选择的值是连续的若干项,所以可以简单的写成:

\[P(x)=\sum_{i=1}^{k+1}P(x_i)\frac{(-1)^{k+1-i}}{(i-1)!(k+1-i)!}\prod_{j=1,j\neq i}^{k+1}(x-x_j)
\]

在这题里,我们都已经知道\(x\)是\(n\)了,所以后半部分的\(prod\)可以用前后缀的方式快速预处理出来,这样子我们就可以\(O(k)\)的计算前面的部分了。

#include<iostream>
#include<cstdio>
using namespace std;
#define MOD 998244353
#define MAX 10000100
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
int n,K,N,ans;
int inv[MAX],jv[MAX],suf[MAX],pre[MAX];
bool zs[MAX];
int pri[MAX],tot,pw[MAX];
int fpow(int a,int b){int s=1;while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}return s;}
void Sieve(int n)
{
pw[1]=1;
for(int i=2;i<=n;++i)
{
if(!zs[i])pri[++tot]=i,pw[i]=fpow(i,K);
for(int j=1;j<=tot&&i*pri[j]<=n;++j)
{
zs[i*pri[j]]=true;
pw[i*pri[j]]=1ll*pw[i]*pw[pri[j]]%MOD;
if(i%pri[j]==0)break;
}
}
}
int P[MAX];
int calc(int n)
{
int ret=0;pre[0]=suf[N+1]=1;
for(int i=1;i<=N;++i)pre[i]=1ll*pre[i-1]*(n-i+MOD)%MOD;
for(int i=N;i;--i)suf[i]=1ll*suf[i+1]*(n-i+MOD)%MOD;
for(int i=1,d=((N+1)&1)?MOD-1:1;i<=N;++i,d=MOD-d)
ret=(ret+1ll*P[i]*d%MOD*jv[i-1]%MOD*jv[N-i]%MOD*pre[i-1]%MOD*suf[i+1])%MOD;
return ret;
}
int main()
{
n=read();K=read();N=K+3;
//for(int i=1;i<=n;++i)ans=(ans+1ll*(i-1)*fpow(i,K))%MOD;
//for(int i=n+1;i<=n+n;++i)ans=(ans+1ll*(n+n-i+1)*fpow(i,K))%MOD;
//for(int i=1;i<=n;++i)ans=(ans+MOD-fpow(2*i,K))%MOD;
Sieve(N);inv[0]=inv[1]=jv[0]=1;
for(int i=2;i<=N;++i)inv[i]=1ll*inv[MOD%i]*(MOD-MOD/i)%MOD;
for(int i=1;i<=N;++i)jv[i]=1ll*jv[i-1]*inv[i]%MOD;
for(int i=1;i<=N;++i)P[i]=(P[i-1]+1ll*(i-1)*pw[i])%MOD;
ans=(ans+calc(n))%MOD;
for(int i=1;i<=N;++i)P[i]=(P[i-1]+1ll*(0ll+n+n-i+1)*pw[i])%MOD;
ans=(ans+calc((n+n)%MOD))%MOD;
ans=(ans+MOD-calc(n))%MOD;
for(int i=1;i<=N;++i)P[i]=(P[i-1]+1ll*pw[i]*pw[2])%MOD;
ans=(ans+MOD-calc(n))%MOD;
ans=1ll*ans*fpow(n,MOD-2)%MOD;
printf("%d\n",ans);
return 0;
}

【洛谷5437】【XR-2】约定(拉格朗日插值)的更多相关文章

  1. [洛谷P4781]【模板】拉格朗日插值

    题目大意:给你$n(n\leqslant2000)$个点,要你求$n-1$次经过这$n$个点的多项式在$k$处的值 题解:$Lagrange$插值:$$f_x=\sum\limits_{i=1}^ky ...

  2. 洛谷P4781 【模板】拉格朗日插值(拉格朗日插值)

    题意 题目链接 Sol 记得NJU有个特别强的ACM队叫拉格朗,总感觉少了什么.. 不说了直接扔公式 \[f(x) = \sum_{i = 1}^n y_i \prod_{j \not = i} \f ...

  3. 洛谷P5437/5442 约定(概率期望,拉格朗日插值,自然数幂)

    题目大意:$n$ 个点的完全图,点 $i$ 和点 $j$ 的边权为 $(i+j)^k$.随机一个生成树,问这个生成树边权和的期望对 $998244353$ 取模的值. 对于P5437:$1\le n\ ...

  4. 洛谷 P7116 - [NOIP2020] 微信步数(拉格朗日插值)

    洛谷题面传送门 我竟然独立切掉了这道题!incredible! 纪念我逝去的一上午(NOIP 总时长 4.5h,这题做了我整整 4.5h) 首先讲一下现场我想的 80 分的做法,虽然最后挂成了 65 ...

  5. 洛谷 P5469 - [NOI2019] 机器人(区间 dp+拉格朗日插值)

    洛谷题面传送门 神仙题,放在 D1T2 可能略难了一点( 首先显然对于 P 型机器人而言,将它放在 \(i\) 之后它会走到左边第一个严格 \(>a_i\) 的位置,对于 Q 型机器人而言,将它 ...

  6. 洛谷 P3270 - [JLOI2016]成绩比较(容斥原理+组合数学+拉格朗日插值)

    题面传送门 考虑容斥.我们记 \(a_i\) 为钦定 \(i\) 个人被 B 神碾压的方案数,如果我们已经求出了 \(a_i\) 那么一遍二项式反演即可求出答案,即 \(ans=\sum\limits ...

  7. LOJ 2743(洛谷 4365) 「九省联考 2018」秘密袭击——整体DP+插值思想

    题目:https://loj.ac/problem/2473 https://www.luogu.org/problemnew/show/P4365 参考:https://blog.csdn.net/ ...

  8. 洛谷P5158 【模板】多项式快速插值

    题面 传送门 前置芝士 拉格朗日插值,多项式多点求值 题解 首先根据拉格朗日插值公式我们可以暴力\(O(n^2)\)插出这个多项式,然而这显然是\(gg\)的 那么看看怎么优化,先来看一看拉格朗日插值 ...

  9. 【Luogu4781】【模板】拉格朗日插值

    [Luogu4781][模板]拉格朗日插值 题面 洛谷 题解 套个公式就好 #include<cstdio> #define ll long long #define MOD 998244 ...

随机推荐

  1. Java开发桌面程序学习(六)——拖动文件获得文件路径

    拖动获得文件路径 在windows软件中,很多软件都提供了拖动文件的打开文件的功能,JavaFx中也是有这功能,是通过监听器来实现的 监听器 setOnDragDetected(new EventHa ...

  2. 百度Sitemap生成器

    今天用了两个小时, 为无限影视(https://www.88tv.org)开发了一个小工具, 用来生成baidu的sitemap.  方便用. 因为该电影站的视频内容详情网页的ID是自增长的,所以可以 ...

  3. 如何访问到静态的文件,如jpg,js,css.

    如果你的DispatcherServlet拦截"*.do"这样的有后缀的URL,就不存在访问不到静态资源的问题. 如果你的DispatcherServlet拦截"/&qu ...

  4. Linux帮助——常用命令

    Linux帮助——常用命令 摘要:本文主要学习了Linux系统中常用的一些命令. uname命令 uname命令可以显示电脑以及操作系统的相关信息. 基本语法 uname [选项] 选项说明 -a:显 ...

  5. 盘点10个CAD难点,看看有没有让你崩溃的,解决方法一并奉上

    蜀道难,难于上青天”,对于很多学习CAD的小伙伴来说CAD就跟蜀道一样,太难了,下面小编分享几个在学习CAD过程中会遇到的问题以及解决的方法,一起来看看吧! 1. 如何替换找不到的原文字体? 答:复制 ...

  6. 敏捷软件开发_实例1<二>

    敏捷软件开发_实例1 这本书的实例非常好,给了我非常多的启发.主要讲了两个实例,咖啡机和薪水支付实例,咖啡机实例比较简单并没有用什么设计模式,薪水支付实例用了很多设计模式,包括后面的打包等. 咖啡机实 ...

  7. 高强度学习训练第十一天总结:Class文件结构(二)

    常量池 可以理解为Class文件之中的资源仓库,他是Class文件结构中与其他项目关联最多的数据类型,也是占用Class文件空间最大的数据项目之一 访问标志 在常量池结束后,紧接着的俩个字节代表访问标 ...

  8. Model赋值返回json

    DataTable resultList = bll.GetResultListByCondition(bureauCode, deptCode, fileTitle); IList<GanBu ...

  9. SSRS 关于日期参数的范围限制

    在进行SSRS Report开发的时候,我们往往会有日期\时间范围限制的需求,但参数的报表参数并没有相关的事件\属性来设置. 所以,我们需要曲线救国. 这里要说的这种方法,仅支持Microsoft S ...

  10. CocoPods原理

    CocoaPods 的原理是将所有的依赖库都放到另一个名为Pods的项目中, 然而让住项目依赖Pods项目, 这样,源码管理工作任务从主项目移到了Pods项目中. 1.Pods项目最终会编译成一个名为 ...