【NOI2020】命运【树形dp】【线段树合并】

题意:给一棵 nnn 个点的树,每条边需要染成黑白两种颜色中的一种。给出 mmm 个条件,每个条件给出 u,vu,vu,v,其中 uuuvvv 的祖先,要求 uuuvvv 的链上至少一条黑边。求方案数 模 998244353998244353998244353

n,m≤5×105n,m\leq 5\times 10^5n,m5×105

这个dp想了一上午

对于树上的一个点,考虑子树内有关的所有限制,唯一不好处理的是超出子树的部分,而这部分只需要考虑超出最短的。

定义: dp(u,k)dp(u,k)dp(u,k) 表示有多少种确定 uuu 子树内的边的颜色的方案,使得所有下端点在 uuu 子树内并且尚未满足的条件 的上端点的深度最大值恰好为 kkk。如果所有上述条件都满足, k=0k=0k=0

人话翻译:

考虑 uuu 子树内能影响到的条件,分为下列两种:

  1. 上端点在子树内(显然下端点就在子树内了)。如果这种条件没有满足,就永远不可能满足了,这个时候上面的定义表现为 k≥depuk\geq dep_ukdepu,后面可以看到这部分状态是无用的。
  2. 上端点是 uuu 的严格祖先,下端点在 uuu 子树内,且 uuu 到下端点这段没有黑边。此时就需要上端点到 uuu 有黑边。如果这样的条件的上端点的最大深度为 kkk,那么所有条件成立当且仅当 uuu 深度为 kkk 的祖先到 uuu 有一条黑边,处理方式后述。

进行一次 dfs,每个点 uuu 先假设它没有儿子,即让 dp(u,x)=1dp(u,x)=1dp(u,x)=1,其中 xxx 为所有下端点为 uuu 的条件的上端点的最大深度。

然后依次突然加入每个儿子,设儿子为 vvv,得到新的 dp 数组为 dp′dp'dp

考虑连接儿子的这条边是黑边还是白边。

如果是黑边,对于 uuu 来说,从 vvv 子树内来的条件就全部满足了(当然要原来有机会满足),但 uuu 原来不满足的还是不满足。即

dp(u,k)∑i=0depudp(v,i)dp(u,k)\sum_{i=0}^{dep_u}dp(v,i)dp(u,k)i=0depudp(v,i)

如果是白边,那么要同时满足两边的深度限制,即

∑max⁡(i,j)=kdp(u,i)dp(v,j)\sum_{\max(i,j)=k}dp(u,i)dp(v,j)max(i,j)=kdp(u,i)dp(v,j)

整理一下

dp′(u,k)=dp(u,k)∑i=0depudp(v,i)+dp(u,k)∑i=0kdp(v,i)+dp(v,k)∑i=0k−1dp(u,i)dp'(u,k)=dp(u,k)\sum_{i=0}^{dep_u}dp(v,i)+dp(u,k)\sum_{i=0}^kdp(v,i)+dp(v,k)\sum_{i=0}^{k-1}dp(u,i)dp(u,k)=dp(u,k)i=0depudp(v,i)+dp(u,k)i=0kdp(v,i)+dp(v,k)i=0k1dp(u,i)

长这样子的式子都可以考虑线段树合并。

∑i=0depudp(v,i)\sum_{i=0}^{dep_u}dp(v,i)i=0depudp(v,i) 是个常数,先算出来。

合并的时候顺便维护左边遍历过的结点的和,如果一边的结点为空,用维护的和给另一边的结点打乘法标记。递归到叶结点了再处理求和符号的边界情况。

注意维护的这个和是合并前的,要先维护再打标记。可以在递归的时候传引用。

复杂度 O(nlog⁡n+m)O(n\log n+m)O(nlogn+m)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cctype>
#include <vector>
#define MAXN 500005
using namespace std;
inline int read()
{int ans=0;char c=getchar();while (!isdigit(c)) c=getchar();while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();return ans;
}
typedef long long ll;
const int MOD=998244353;
inline int add(const int& x,const int& y){return x+y>=MOD? x+y-MOD:x+y;}
int n;
int ch[MAXN<<5][2],sum[MAXN<<5],mul[MAXN<<5],cnt;
inline void update(int x){sum[x]=add(sum[ch[x][0]],sum[ch[x][1]]);}
inline void pushlzy(int x,ll v){sum[x]=sum[x]*v%MOD,mul[x]=mul[x]*v%MOD;}
inline void pushdown(int x)
{if (mul[x]!=1){if (ch[x][0]) pushlzy(ch[x][0],mul[x]);if (ch[x][1]) pushlzy(ch[x][1],mul[x]);mul[x]=1;}
}
void modify(int& x,int l,int r,int k)
{if (!x) mul[x=++cnt]=1;if (l==r) return (void)(++sum[x]);int mid=(l+r)>>1;if (k<=mid) modify(ch[x][0],l,mid,k);else modify(ch[x][1],mid+1,r,k);update(x);
}
int query(int x,int l,int r,int ql,int qr)
{if (!x) return 0;if (ql<=l&&r<=qr) return sum[x];if (qr<l||r<ql) return 0;pushdown(x);int mid=(l+r)>>1;return add(query(ch[x][0],l,mid,ql,qr),query(ch[x][1],mid+1,r,ql,qr));
}
int merge(int x,int y,int l,int r,int& xsum,int& ysum)
{if (!x&&!y) return 0;if (!x) return ysum=add(ysum,sum[y]),pushlzy(y,xsum),y;if (!y) return xsum=add(xsum,sum[x]),pushlzy(x,ysum),x;if (l==r){ysum=add(ysum,sum[y]);int t=sum[x];sum[x]=((ll)sum[x]*ysum+(ll)xsum*sum[y])%MOD;xsum=add(xsum,t);return x;	} pushdown(x),pushdown(y);int mid=(l+r)>>1;ch[x][0]=merge(ch[x][0],ch[y][0],l,mid,xsum,ysum);ch[x][1]=merge(ch[x][1],ch[y][1],mid+1,r,xsum,ysum);update(x);return x;
}
vector<int> e[MAXN],lis[MAXN];
int dep[MAXN],rt[MAXN];
void dfs(int u)
{int mx=0;for (int i=0;i<(int)lis[u].size();i++) mx=max(mx,dep[lis[u][i]]);modify(rt[u],0,n,mx);for (int i=0;i<(int)e[u].size();i++)if (!dep[e[u][i]]){dep[e[u][i]]=dep[u]+1;dfs(e[u][i]);int xsum=0,ysum=query(rt[e[u][i]],0,n,0,dep[u]);rt[u]=merge(rt[u],rt[e[u][i]],0,n,xsum,ysum);}
}
int main()
{n=read();for (int i=1;i<n;i++){int u,v;u=read(),v=read();e[u].push_back(v),e[v].push_back(u);}int m=read();while (m--){int u,v;u=read(),v=read();lis[v].push_back(u);}dfs(dep[1]=1);printf("%d\n",query(rt[1],0,n,0,0));return 0;
}

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

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

相关文章

.NET Core 3.0之深入源码理解Kestrel的集成与应用(二)

前言前一篇文章主要介绍了.NET Core继承Kestrel的目的、运行方式以及相关的使用&#xff0c;接下来将进一步从源码角度探讨.NET Core 3.0中关于Kestrel的其他内容&#xff0c;该部分内容&#xff0c;我们无需掌握&#xff0c;依然可以用好Kestrel&#xff0c;本文只是将一些内部…

Codeforces Round #726 (Div. 2) E2. Erase and Extend (Hard Version) 贪心

传送门 文章目录题意&#xff1a;思路&#xff1a;题意&#xff1a; 给你一个长度为nnn的串sss&#xff0c;你有两个操作可以使用&#xff1a; (1)(1)(1)从sss的结尾删除一个字母。 (2)sss(2)sss(2)sss。 让你通过若干次操作使其变成一个长度为kkk的串&#xff0c;且其字典序最…

【BZOJ3451】Normal【期望线性性】【点分治】【NTT卷积】

题意&#xff1a;随机分治中心点分治的期望操作次数 n≤3104n\leq 3\times 10^4n≤3104 即求点分树的 siz 之和的期望 即祖孙关系对数期望 考虑一有序点对 (u,v)(u,v)(u,v)&#xff0c;uuu 在点分树上是 vvv 祖先当且仅当 uuu 是 u∼vu\sim vu∼v 路径上第一个被选为分治中心…

译 | 使用Roslyn分析器高效编写更好的代码

原文&#xff1a;Mika Dumont翻译&#xff1a;汪宇杰Roslyn 是 .NET 编译器平台&#xff0c;即使在运行代码之前&#xff0c;它也能帮助您捕获 Bug。例如内置于 Visual Studio 中的Roslyn 拼写检查分析器。假设您正在创建一个静态方法&#xff0c;并将 static 单词拼错为 statc…

Codeforces Round #724 (Div. 2) F. Omkar and Akmar 组合数学 + 博弈

传送门 文章目录题意&#xff1a;思路&#xff1a;题意&#xff1a; 思路&#xff1a; 首先我们先来研究一下这个游戏&#xff0c;手画几个会惊奇的发现&#xff0c;后手这个b怎么怎么画都赢啊&#xff1f;&#xff1f;&#xff1f;对&#xff0c;没错&#xff0c;就是怎么画都…

Visual Studio 支持 Java?谣言止于智者

在去年的某一段时间内&#xff0c;我们经常会看到标题类似为“宇宙第一 IDE Visual Studio 支持 Java 了&#xff01;”的文章。当时&#xff0c;韩老师看到这篇文章也是一脸懵逼。公司里做 Java 的开发童鞋们就坐在韩老师附近&#xff08;嗯&#xff0c;不超过五米的距离&…

【SCOI2014】方伯伯的商场之旅【数位dp】【单峰函数】

题意&#xff1a;给定 l,r,kl,r,kl,r,k &#xff0c;对于一个 kkk 进制数&#xff0c;将数码看成这个位置的石子个数&#xff0c;每将一个石子移动 111 的距离需要 111 的代价。求 [l,r][l,r][l,r] 中的所有数在 kkk 进制下将石子集中在一个位置的最小代价之和。 l≤r≤1015,k≤…

Codeforces Round #597 (Div. 2) D. Shichikuji and Power Grid 最小生成树 + 虚拟点

传送门 文章目录题意&#xff1a;思路&#xff1a;题意&#xff1a; 思路&#xff1a; 看错题导致误入歧途&#xff0c;如果能早点看见翻译也不至于一天多也没想出来。 求联通的最小代价&#xff0c;自然的想到了能不能建边跑最小生成树。 对于两点之间比较好弄&#xff0c;直…

【NOI2007】货币兑换【任意坐标斜率优化】【CDQ分治】

题意&#xff1a;有 A&#xff0c;B 两种金券&#xff0c;给出 nnn 天内分别的单位价格和可以购买的数量的比例。开始有 SSS 元&#xff0c;求 nnn 天后最多能有多少元。 提示&#xff1a;每次操作一定全买全卖 n≤105n\leq 10^5n≤105 设 fnf_nfn​ 表示第 nnn 天结束后手上…

小白开学Asp.Net Core 《八》

小白开学Asp.Net Core 《八》— — .Net Core 数据保护组件1、背景我在搞&#xff08;https://github.com/AjuPrince/Aju.Carefree&#xff09;这个开源项目的时候&#xff0c;想做一些防止恶意攻击的小功能&#xff08;如果 我通过页面 /Dome/GetData?id123&#xff0c;那是不…

你必须知道的Docker镜像仓库的搭建

近期工作中发现用到的容器镜像越来越多&#xff08;不多的时候没考虑过镜像仓库的问题&#xff09;&#xff0c;同一个容器镜像也存在多个版本&#xff0c;那么镜像仓库的搭建需求就涌现出来&#xff0c;本文就目前的几个常用镜像仓库的搭建进行介绍&#xff0c;我们可以根据需…

nowcoder 河 我 车 题 错 天 乐 赛 倍增处理

传送门 文章目录题意&#xff1a;思路&#xff1a;题意&#xff1a; 思路&#xff1a; 倍增经典题了&#xff0c;考虑维护一个前缀和sum[i]sum[i]sum[i]&#xff0c;对于iii位置我们肯定是贪心的选能到的最远位置&#xff0c;即第一个>sum[i−1]k>sum[i-1]k>sum[i−1…

狄利克雷生成函数

一个没啥用的东西&#xff0c;权当加深对数论函数和狄利克雷卷积的理解。 定义 序列 {f1,f2,…}\{f_1,f_2,\dots\}{f1​,f2​,…} 的狄利克雷生成函数 DGF 定义为 f(s)∑n1∞fnnsf(s)\sum_{n1}^{\infin}\frac{f_n}{n^s}f(s)n1∑∞​nsfn​​ 这里不要求 fff 是积性函数。 显…

ASP.NET Core Web Api之JWT(一)

最近沉寂了一段&#xff0c;主要是上半年相当于休息和调整了一段时间&#xff0c;接下来我将开始陆续学习一些新的技能&#xff0c;比如Docker、Jenkins等&#xff0c;都会以生活实例从零开始讲解起&#xff0c;到时一并和大家分享和交流。接下来几节课的内容将会讲解JWT&#…

Codeforces Round #658 (Div. 2) D. Unmerge 思维 + 背包

传送门 文章目录题意&#xff1a;思路&#xff1a;题意&#xff1a; 思路&#xff1a; 我们可以假设有两个桶&#xff0c;让后向里面放数&#xff0c;可以发现&#xff0c;如果向其中放入了aia_iai​&#xff0c;那么紧跟在他后面的所有<ai<a_i<ai​的数都必须跟在ai…

【牛客NOIP模拟】牛半仙的魔塔(增强版)【贪心】【并查集】

题意&#xff1a;一个魔塔游戏的地图是一棵以 111 为根的树&#xff0c;起点为根&#xff0c;除根外每个结点有一个怪物&#xff0c;给定每个怪物血量、攻击、防御、奖励蓝宝石个数&#xff08;加防御&#xff09;&#xff0c;勇士的血量、攻击、防御&#xff0c;遇到怪物必须战…

撸过一万行代码,你看过这篇文章吗?

北上广深真的是技术的宝地吗&#xff1f;犹记得刚刚参加工作之后&#xff0c;有人邀请我去深圳&#xff0c;说沿海地区是IT行业从业人员的宝地&#xff0c;只有在那里才会学到真正的技术&#xff0c;而且能拿到更高的工资&#xff0c;不管从哪个层面来说&#xff0c;都值得你去…

Codeforces Round #727 (Div. 2) E. Game with Cards dp + 思维

传送门 文章目录题意&#xff1a;思路&#xff1a;题意&#xff1a; 初始有左右手&#xff0c;上面各有一个数字为000的卡牌&#xff0c;每次都有一个新卡kik_iki​&#xff0c;你可以将其放在左手或者右手&#xff0c;使两只手上的卡片范围在[ll,i,rl,i][l_{l,i},r_{l,i}][ll…

【CEOI2017】Building Bridges【任意坐标斜率优化】【李超线段树】

题意&#xff1a;有 nnn 个柱子&#xff0c;每个柱子有高度 hih_ihi​。你需要在柱子间修桥&#xff0c;在 i,ji,ji,j 间修桥代价为 (hi−hj)2(h_i-h_j)^2(hi​−hj​)2,桥梁只能在柱子处相交&#xff0c;未安装桥的柱子需要拆除&#xff0c;代价为 wiw_iwi​&#xff08;可能为…

【牛客NOIP模拟】路径难题【建图】【最短路证明】

题意&#xff1a;一张 nnn 个点 mmm 条边的无向图&#xff0c;边带距离&#xff0c;可以坐出租车&#xff0c;花费为距离除以常数 rrr 向上取整&#xff1b;也可以坐公交车&#xff0c;每路车行驶路线给定&#xff0c;无论坐多少站花费都为 cic_ici​ &#xff08;每路车可能不…