[CF995F] Cowmpany Cowmpensation(树形dp,拉格朗日插值)

树形DP:
f[u][i]f[u][i]f[u][i]表示给uuu的子树分配工资,uuu点工资为iii的方案数
f[u][i]=∏v∈sonu(∑j=1if[v][j])f[u][i]=\prod\limits_{v\in son_u}(\sum\limits_{j=1}^{i}f[v][j])f[u][i]=vsonu(j=1if[v][j])
前缀和优化:
g[u][i]=∑j=1if[u][j]g[u][i]=\sum\limits_{j=1}^{i}f[u][j]g[u][i]=j=1if[u][j]
f[u][i]=∏v∈sonug[v][i]f[u][i]=\prod\limits_{v\in son_u}g[v][i]f[u][i]=vsonug[v][i]
时间复杂度O(nd)O(nd)O(nd)
考虑用拉格朗日插值优化O(n2)O(n^2)O(n2)
gu(x)g_u(x)gu(x)为关于xxx的函数,代入xxx,即可得到g[u][x]g[u][x]g[u][x]
合理猜想gu(x)g_u(x)gu(x)的次数为szusz_uszuuuu的子树大小),可以用数学归纳法证明:

  • uuu 为叶子结点时,gu(x)=xg_u(x)=xgu(x)=x,成立。
  • uuu 非叶子结点时,考虑:
    gu(x)−gu(x−1)=∏v∈sonugv(x)g_u(x)-g_u(x-1)=\prod\limits_{v\in son_u}g_v(x)gu(x)gu(x1)=vsonugv(x)
    由于 vvv 满足猜想,即gv(x)g_v(x)gv(x) 的次数为 szvsz_vszv,则 gu(x)−gu(x−1)g_u(x)-g_u(x-1)gu(x)gu(x1) 的次数为 ∑v∈sonuszv\sum\limits_{v\in son_u}sz_vvsonuszv,即 szu−1sz_u-1szu1
    再还原差分,次数 +1,有 gu(x)g_u(x)gu(x) 是关于 xxxszu−1+1=szusz_u-1+1=sz_uszu1+1=szu 次函数。

所以我们只要知道g1(1),g1(2),...,g1(n)g_1(1),g_1(2),...,g_1(n)g1(1),g1(2),...,g1(n),便可以用拉格朗日插值得出g1(d)g_1(d)g1(d)

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=3050;
struct Edge{int v,nxt;}edge[N];
int n,cnt,head[N],fa[N];
int d,f[N][N],sum[N][N],inv[N],ans;
int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
int mul(int a,int b){return 1ll*a*b%mod;}
void addedge(int u,int v){edge[++cnt].v=v;edge[cnt].nxt=head[u];head[u]=cnt;
}
void dfs(int u){for(int i=head[u];i;i=edge[i].nxt){int v=edge[i].v;dfs(v);for(int j=1;j<=min(n,d);j++)f[u][j]=mul(f[u][j],sum[v][j]);}for(int j=1;j<=min(n,d);j++) sum[u][j]=add(sum[u][j-1],f[u][j]);
}
int main(){scanf("%d%d",&n,&d);inv[1]=1;for(int i=2;i<=n;++i) inv[i]=mul(mod-mod/i,inv[mod%i]);for(int i=2;i<=n;i++){scanf("%d",&fa[i]);addedge(fa[i],i);}for(int i=1;i<=n;i++)for(int j=1;j<=min(n,d);j++) f[i][j]=1;dfs(1);if(d<=n) printf("%d\n",sum[1][d]);else{for(int i=1;i<=n;i++){int x=sum[1][i];for(int j=0;j<=n;++j) if(i^j) x=1ll*x*(d-j+mod)%mod*(i>j?inv[i-j]:mod-inv[j-i])%mod;ans=add(ans,x);}printf("%d\n",ans);}return 0;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/319844.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【dfs】益智游戏(2017 特长生 T2)

题目大意 给你四个数字&#xff0c;问你能否经过加减乘除使其结果为24 解题思路 先暴力枚举四个数字的全排列&#xff0c;然后枚举运算符和括号 代码 #include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define ll long …

分布式系统消息中间件——RabbitMQ的使用基础篇

前言我是在解决分布式事务的一致性问题时了解到RabbitMQ的&#xff0c;当时主要是要基于RabbitMQ来实现我们分布式系统之间对有事务可靠性要求的系统间通信的。关于分布式事务一致性问题及其常见的解决方案&#xff0c;可以看我另一篇博客。提到RabbitMQ&#xff0c;不难想到的…

P4630-[APIO2018]Duathlon铁人两项【圆方树】

正题 题目链接:https://www.luogu.com.cn/problem/P4630 题目大意 nnn个点mmm条边的一张无向图&#xff0c;求有多少对三元组(s,c,f)(s,c,f)(s,c,f)满足s≠f≠ts\neq f\neq ts​f​t且存在一条从sss到fff的简单路径经过ccc 解题思路 一个比较显然的结论是在一个点双中的三…

小a的旅行计划

来源&#xff1a;牛客网 时间限制&#xff1a;C/C 1秒&#xff0c;其他语言2秒 空间限制&#xff1a;C/C 32768K&#xff0c;其他语言65536K 64bit IO Format: %lld文章目录题目描述题解&#xff1a;代码&#xff1a;题目描述 小a终于放假了&#xff0c;它想在假期中去一些地方…

吃豆人(luogu 7472/NOI Online 2021 普及组 T2)

正题 luogu 7472 题目大意 给出一个正方形点阵&#xff0c;让你选择两个点&#xff0c;分别向两个方向移动&#xff08;必须是45度&#xff09;&#xff0c;每到一个点就得到该点的贡献&#xff08;不重复得&#xff09;&#xff0c;遇到墙壁反射&#xff0c;问你最大贡献 解…

[XSY] 选举(线段树优化dp)

选举 //a[i]1 or 0 or -1,表示i支持小奇 or 中立 or 支持魔法猪 //sum[i]a[1]a[2]...a[i] //设dp[i]表示把1~i划分成若干区间,(小奇得票-魔法猪得票)的最大值 /*dp[i]max {dp[j]1 (sum[i]-sum[j]>0即sum[i]>sum[j])dp[j] (sum[i]-sum[j]0即sum[i]sum[j])dp[j]-1 (sum[…

P3480-[POI2009]KAM-Pebbles【阶梯博弈】

正题 题目链接:https://www.luogu.com.cn/problem/P3480 题目大意 nnn个石头堆上进行Nim\text{Nim}Nim游戏&#xff0c;不过需要满足每次操作前后都有ai≤ai1(i∈[1,n))a_i\leq a_{i1}(\ i\in[1,n)\ )ai​≤ai1​( i∈[1,n) ) 解题思路 让每一个biai−ai−1b_ia_i-a_{i-1}bi​…

在 Centos7 用Jexus服务器 运行.Net Core 只需几步

安装 .Net SDK 不需要按照 .net core runtime,sdk 依赖于运行时会自动安装第一步 添加dotnet源sudo rpm -Uvh https://packages.microsoft.com/config/rhel/7/packages-microsoft-prod.rpm第二步 安装 .Net Core可能由于网络原因&#xff0c;下载要耐心等待一段时间,下载完成后…

小球碰撞(理解ing)

来源&#xff1a;牛客网&#xff1a; 文章目录题目描述题解&#xff1a;代码&#xff1a;时间限制&#xff1a;C/C 1秒&#xff0c;其他语言2秒 空间限制&#xff1a;C/C 262144K&#xff0c;其他语言524288K 64bit IO Format: %lld题目描述 一个弹球&#xff08;可视为质点&am…

【斜率优化】Cats Transport(luogu-CF 311B)

正题 luogu-CF 311B 题目大意 有n个点&#xff0c;p个铲屎官和m只猫 did_idi​表示i到i-1的距离 第i只猫在tit_iti​这个时间hih_ihi​这个点开始等待铲屎官来接它走 铲屎官会以一个单位时间走一个单位长度的速度从第一个点向最后一个点走&#xff08;不能停止&#xff09;&a…

[XSY4170] 妹子(线段树上二分)

传送门 题意&#xff1a; 给出两个长度为NNN的整数序列A1A2...ANA_1A_2...A_NA1​A2​...AN​,B1B2...BNB_1B_2...B_NB1​B2​...BN​。有MMM组询问(opt,l,r,d)(opt,l,r,d)(opt,l,r,d)&#xff0c;令nr−l1nr-l1nr−l1 若optAoptAoptA&#xff1a;∀i∈[1,n],aiAli−1d,biBli−…

P3643-[APIO2016]划艇【dp】

正题 题目链接:https://www.luogu.com.cn/problem/P3643 题目大意 求有多少个nnn个数的序列xxx满足&#xff0c;xi∈{0}∪[ai,bi]x_i\in \{0\}\cup[a_i,b_i]xi​∈{0}∪[ai​,bi​]且非000数递增。 解题思路 会发现ai,bia_i,b_iai​,bi​很大不能太暴力的将第二维的dpdpdp设…

.Net思想篇:为何我们需要思想大洗礼?

前言每当说一件事的时候&#xff0c;我喜欢回溯往事&#xff0c;这样思有所依&#xff0c;文有所凭&#xff0c;才能达到打字如尿崩&#xff0c;一发不可收拾的流畅度。让我们将时光回溯到08年&#xff0c;当时和同学们就有了学java和.net之争&#xff0c;当然不止这些&#xf…

【Manacher】最长双回文串(luogu 4555)

正题 luogu 4555 题目大意 给出一个字符串&#xff0c;让你求两个相邻的回文串的最大长度和 解题思路 先用Manacher求出最长回文串 然后暴力配对即可 代码 #include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define l…

[AGC009B] Tournament(多叉树转二叉树后的最小可能深度)

传送门 把aia_iai​看成faifa_ifai​&#xff0c;建出一棵多叉树&#xff0c;再把多叉树转成二叉树&#xff0c;转出来的每棵二叉树对应着一种比赛方式。 以n8,a2,...,81,1,2,4,3,3,3n8,a_{2,...,8}1,1,2,4,3,3,3n8,a2,...,8​1,1,2,4,3,3,3为例&#xff0c; 多叉树转出的二…

C# 8中的Async Streams

关键要点异步编程技术提供了一种提高程序响应能力的方法。Async/Await模式在C# 5中首次亮相&#xff0c;但只能返回单个标量值。C# 8添加了异步流&#xff08;Async Streams&#xff09;&#xff0c;允许异步方法返回多个值&#xff0c;从而扩展了其可用性。异步流提供了一种用…

线段树回顾

这个博主的线段树我觉得讲的很细了 文章目录建树区间查询&#xff0c;单点修改区间修改&#xff0c;单点查询区间修改&#xff0c;区间查询&#xff08;带pushdown&#xff09;乘法线段树根号线段树建树 struct node{ll l,r;ll sum,mlz,plz; }tree[4*maxn]; inline void bulid…

P3175-[HAOI2015]按位或【min-max容斥,FWT】

正题 题目链接:https://www.luogu.com.cn/problem/P3175 题目大意 开始有一个nnn位二进制数s0s0s0&#xff0c;每次有pip_ipi​概率选取数字iii让sss或上这个数字iii&#xff0c;求期望多少次能够让sss的nnn个位都变为111。 解题思路 因为是或所以我们只关心最后一个选中的数…

【DP】建学校问题(luogu 2803/2018 特长生 T3)

正题 luogu 2803 题目大意 给出n个点&#xff0c;和相邻的点的距离&#xff0c;每个点有一个权值&#xff0c;现在让你建k个特殊点&#xff0c;使所有点到其中一个特殊点的代价之和最小 解题思路 先预处理处一个区间到同一个特殊点的最小代价 然后DP即可 – 代码 #inclu…

[XSY3112] 接水果(树上包含路径,整体二分,扫描线)

传送门 给出一棵nnn个点的树。接下来给出PPP条树上路径ai→bia_i\to b_iai​→bi​&#xff0c;及其权值cic_ici​。最后有QQQ个询问&#xff0c;每个询问给出一条树上路径ui→viu_i\to v_iui​→vi​&#xff0c;问在包含ui→viu_i\to v_iui​→vi​的所有树上路径中&#xf…