其他分享
首页 > 其他分享> > NOI 2020 D1T2 命运(Destiny) Solution

NOI 2020 D1T2 命运(Destiny) Solution

作者:互联网

NOI 2020 D1T2 Destiny

Task

给定 \(n\) 个点的以 \(1\) 为根的有根树,有 \(m\) 条约束,每条约束包含一个点对 \((u,v)\),满足 \(u\) 是 \(v\) 的祖先,你需要给每条边染成黑白两种颜色,满足对于每条约束,\(u\rightarrow v\) 的路径上都有一条黑边,求合法方案数。

答案对 \(998244353\) 取模。

限制:\(1\le n,m\le 5\times10^5\)

Solution

预处理 \(\text{limit}[u]\) 表示 \(u\) 往上的黑边中,最近的黑边的深度需要大于等于 \(\text{limit}[u]\)。这里的 \(\text{limit}\) 仅考虑以点 \(u\) 为下端点的限制

用 \(\text{parent[u]}\) 来表示 \(u\) 的父亲结点,设 \(\text{dp}[u][d]\) 表示 \(\text{parent}[u]\) 往上最近的黑边的深度为 \(d\) 的方案数。

设 \(\mathcal V\) 表示点 \(u\) 的儿子集合。

接下来考虑求出 \(\text{dp}[u][d]\),根据边 \((\text{parent}[u],u)\) 染色的情况分类讨论:

\[\text{dp}[u][d]=[d\ge\text{limit[u]}]\prod_{v\in\mathcal V}\text{dp}[v][d]+\prod_{v\in \mathcal V} \text{dp}[v][\text{dep}[u]] \]

前者是染白的方案数,后者是染黑的方案数。

然后我们考虑对树上的每一个结点开一棵线段树维护 \(\text{dp}\) 数组,具体来说,结点 \(u\) 所对应的线段树的 \([d,d]\) 区间维护了 \(\text{dp}[u][d]\) 的值。

然后我们考虑 \(\text{dp}\) 的转移是如何体现在线段树上的操作的:

  1. 我们将线段树的 \(\text{limit}[u]\sim n\) 的位置区间加为 \(1\),表示这些位置是可以被染成白边的。
  2. 我们将 \(u\) 的所有儿子对应的线段树合并起来,也就是对应位置的值相乘,这一步是为了计算染成白边的方案数。
  3. 接下来我们查询这棵树上 \(\text{dep}[u]\) 对应的值,全局加上这个值。

总结一下,我们的线段树需要支持

  1. 区间加
  2. 单点查询
  3. 区间对应位置乘

这里需要使用到线段树合并 pushdown 的 trick,只要该点没有左右儿子,就不需要 pushdown

什么时候,我们才能真正决定自己的命运呢?

标签:NOI,parent,text,线段,Solution,limit,mathcal,dp,Destiny
来源: https://www.cnblogs.com/hl-fc/p/15415834.html