【ICPC2022济南站】【树形dp】【删物品背包dp】C.DFS Order 2
admin
2024-04-15 08:56:30
0

【题意】

题目链接:https://codeforces.com/gym/104076/problem/C
简要题意:给定一棵n个点的有根树,对于所有的二元组(i,j)(i,j)(i,j)求这颗树所有可能的dfs序中有多少个dfs序满足第iii个点出现在dfs序第jjj个位置。

【思路】

赛场上假了无数次以后,我终于才理清楚了这题的dp思路。
状态定义:
定义dp[u][i]dp[u][i]dp[u][i]表示只考虑uuu子树外的点的情况下,dfs序中在uuu前面有iii个点的方案数。注意,这个dpdpdp值并不能直接作为答案,还要乘上uuu子树内部的所有可能的dfs序方案数。显然这个dpdpdp的取值与uuu子树的情况无关,因此这道题dpdpdp的转移与一般树形dpdpdp不同,这道题应当自上而下用父亲的信息更新儿子的信息。上文提到过,为了得到答案,我们还需要uuu子树内部的dfs序方案数量,因此定义dp2[u]dp2[u]dp2[u]表示uuu子树内的dfs序方案数。
状态转移
设我们当前在uuu点,我们考虑更新uuu的一个儿子vvv的dpdpdp信息,我们需要知道dfs序有多少个点在vvv前面,我们把这些点分为在uuu的子树内和uuu的子树外两类,最后类似背包的思路合并即可。

  • 对于uuu子树外的点的信息,我们通过uuu的dpdpdp值即可获得;
  • 对于uuu子树内的点,在vvv前面的点的数量取决于uuu的儿子们的排列顺序,我们可以把u的所有儿子的子树大小sizsizsiz拿来做一次背包,与一般背包不同的是,后续用背包的结果更新dp值时还需要考虑放置顺序,因此我们还需要加一维表示当前背包的大小,便于考虑物品顺序。设c[i][j]c[i][j]c[i][j]表示放置了iii个物品,总大小为jjj的方案数,考虑放置子树vvv,则转移显然:c[i][j]=c′[i][j]+c′[i−1][j−siz[v]]c[i][j]=c'[i][j]+c'[i-1][j-siz[v]]c[i][j]=c′[i][j]+c′[i−1][j−siz[v]]
    c′c'c′表示未考虑vvv的dp值,ccc表示已考虑vvv的dp值,因此iii这一维倒序枚举更新即可不需要额外的数组。若我们要求vvv的dp值,则我们背包里需要删除vvv这个物品。由于背包dp的本质是一种多项式的卷积(此处的dp等价于c(y,x)=∏v∈sonu(1+y∗xsiz[v])c(y,x)=\prod_{v\in son_u}(1+y*x^{siz[v]})c(y,x)=∏v∈sonu​​(1+y∗xsiz[v])),放置顺序无关紧要,我们不妨假设vvv是最后一个加入背包的物品。此时相当于我们已知c数组而需要求c′c'c′数组(注意,这里的c′c'c′数组在参考代码里就是ddd数组)。移项可得转移:
    c′[i][j]=c[i][j]−c′[i−1][j−siz[v]]c'[i][j]=c[i][j]-c'[i-1][j-siz[v]]c′[i][j]=c[i][j]−c′[i−1][j−siz[v]]
    在得到c′c'c′数组后,设在vvv前面有jjj个uuu子树内的点的方案数为b[j]b[j]b[j],uuu一共有cntcntcnt个儿子,则有转移:
    b[j]=∑i=0cnt−1c′[i][j]∗i!∗(cnt−1−i)!b[j]=\sum_{i=0}^{cnt-1} c'[i][j]*i!*(cnt-1-i)!b[j]=i=0∑cnt−1​c′[i][j]∗i!∗(cnt−1−i)!
    而对于uuu这个点,它在dfs序中一定位于其他uuu子树内的点的前面,我们可以在设置ccc数组初值时考虑它,即:c[0][1]=1c[0][1]=1c[0][1]=1。
  • 现在,我们已知bbb数组和dpdpdp数组,考虑用背包dp的思路对它们进行合并。但是我们目前仅考虑了u的儿子之间的排列顺序,尚未考虑这些儿子子树内的排序方案。设B=∏x∈sonudp2[x]B=\prod_{x\in son_u}dp2[x]B=∏x∈sonu​​dp2[x]。当我们用这些信息去更新vvv的dp值时,需要注意除去常数B中和vvv子树有关的信息。诚然,不除去这部分信息的话,我们直接得到的就是vvv的答案数组,但是这并不利于我们进一步dfs求v的子树的dp信息,因此我们在这里的处理是除去这部分信息。即定义C=∏x∈sonu,x≠v=Bdp2[v]C=\prod_{x\in son_u,x\neq v}=\frac B {dp2[v]}C=∏x∈sonu​,x​=v​=dp2[v]B​,有以下转移:
    对于i∈[1,n],dp[v][i]=C∗∑j=0siz[u]b[j]∗dp[u][i−j]对于i\in [1,n],dp[v][i]=C*\sum_{j=0}^{siz[u]}b[j]*dp[u][i-j]对于i∈[1,n],dp[v][i]=C∗j=0∑siz[u]​b[j]∗dp[u][i−j]
    (jjj的上确界是小于siz[u]siz[u]siz[u]的,比赛时为了求稳所以定枚举上界为siz[u]siz[u]siz[u])

而对于dp2数组,我们需要在处理dp数组前就提前dfs一次先得到它。设uuu总共有cntcntcnt个儿子,考虑uuu子树内所有可能的dfs序,转移显然:
dp2[u]=cnt!∗∏dp2[v]dp2[u]=cnt!*\prod dp2[v]dp2[u]=cnt!∗∏dp2[v]
此时,第iii个点出现在dfs序中第jjj个位置的方案数就是dp[i][j−1]∗dp2[i]dp[i][j-1]*dp2[i]dp[i][j−1]∗dp2[i]。
参考代码:(比赛时写的代码)

#include
#define ll long long
using namespace std;
const int N=505,mod=998244353;
int n,dp[N][N],siz[N],c[N][N],d[N][N],b[N],fac[N],dp2[N];
vectorg[N];
void Add(int&x,int y){((x+=y)>=mod)&&(x-=mod);
}
inline int mul(int x,int y){return 1ll*x*y%mod;
}
inline int dec(int x,int y){return xint ret=1;while(b){if(b&1)ret=mul(ret,a);a=mul(a,a);b>>=1;}return ret;
}
void dfs1(int u,int fa){siz[u]=1;dp2[u]=1;int cnt=0;for(int v:g[u]){if(v==fa)continue;cnt++;dfs1(v,u);siz[u]+=siz[v];dp2[u]=mul(dp2[u],dp2[v]);}dp2[u]=mul(dp2[u],fac[cnt]);
}
void dfs2(int u,int fa){int cnt=0,sum=1;memset(c,0,sizeof c);c[0][1]=1;int lsr=1;for(int v:g[u]){if(v==fa)continue;cnt++;sum+=siz[v];lsr=mul(lsr,dp2[v]);for(int i=cnt;i>=1;i--)for(int j=sum;j>=siz[v];j--)Add(c[i][j],c[i-1][j-siz[v]]);}for(int v:g[u]){if(v==fa)continue;memset(d,0,sizeof d);for(int i=0;i<=cnt;i++)for(int j=0;j<=siz[u];j++)d[i][j]=dec(c[i][j],(j>=siz[v]&&i>0)?d[i-1][j-siz[v]]:0);memset(b,0,sizeof b);for(int i=0;iif(v!=fa)dfs2(v,u);}
}
int main()
{scanf("%d",&n);fac[0]=1;for(int i=1;i<=n;i++)fac[i]=mul(fac[i-1],i);for(int i=2;i<=n;i++){int u,v;scanf("%d%d",&u,&v);g[u].push_back(v);g[v].push_back(u);}dp[1][0]=1;dfs1(1,0);dfs2(1,0);for(int i=1;i<=n;i++){for(int j=0;j

相关内容

热门资讯

【MySQL】锁 锁 文章目录锁全局锁表级锁表锁元数据锁(MDL)意向锁AUTO-INC锁...
【内网安全】 隧道搭建穿透上线... 文章目录内网穿透-Ngrok-入门-上线1、服务端配置:2、客户端连接服务端ÿ...
GCN的几种模型复现笔记 引言 本篇笔记紧接上文,主要是上一篇看写了快2w字,再去接入代码感觉有点...
数据分页展示逻辑 import java.util.Arrays;import java.util.List;impo...
Redis为什么选择单线程?R... 目录专栏导读一、Redis版本迭代二、Redis4.0之前为什么一直采用单线程?三、R...
【已解决】ERROR: Cou... 正确指令: pip install pyyaml
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
Lock 接口解读 前置知识点Synchronized synchronized 是 Java 中的关键字,...
Win7 专业版安装中文包、汉... 参考资料:http://www.metsky.com/archives/350.htm...
3 ROS1通讯编程提高(1) 3 ROS1通讯编程提高3.1 使用VS Code编译ROS13.1.1 VS Code的安装和配置...
大模型未来趋势 大模型是人工智能领域的重要发展趋势之一,未来有着广阔的应用前景和发展空间。以下是大模型未来的趋势和展...
python实战应用讲解-【n... 目录 如何在Python中计算残余的平方和 方法1:使用其Base公式 方法2:使用statsmod...
学习u-boot 需要了解的m... 一、常用函数 1. origin 函数 origin 函数的返回值就是变量来源。使用格式如下...
常用python爬虫库介绍与简... 通用 urllib -网络库(stdlib)。 requests -网络库。 grab – 网络库&...
药品批准文号查询|药融云-中国... 药品批文是国家食品药品监督管理局(NMPA)对药品的审评和批准的证明文件...
【2023-03-22】SRS... 【2023-03-22】SRS推流搭配FFmpeg实现目标检测 说明: 外侧测试使用SRS播放器测...
有限元三角形单元的等效节点力 文章目录前言一、重新复习一下有限元三角形单元的理论1、三角形单元的形函数(Nÿ...
初级算法-哈希表 主要记录算法和数据结构学习笔记,新的一年更上一层楼! 初级算法-哈希表...
进程间通信【Linux】 1. 进程间通信 1.1 什么是进程间通信 在 Linux 系统中,进程间通信...
【Docker】P3 Dock... Docker数据卷、宿主机与挂载数据卷的概念及作用挂载宿主机配置数据卷挂载操作示例一个容器挂载多个目...