【NOI2020】命运【树形dp】【线段树合并】
题意:给一棵 nnn 个点的树,每条边需要染成黑白两种颜色中的一种。给出 mmm 个条件,每个条件给出 u,vu,vu,v,其中 uuu 是 vvv 的祖先,要求 uuu 到 vvv 的链上至少一条黑边。求方案数 模 998244353998244353998244353。
n,m≤5×105n,m\leq 5\times 10^5n,m≤5×105
这个dp想了一上午
对于树上的一个点,考虑子树内有关的所有限制,唯一不好处理的是超出子树的部分,而这部分只需要考虑超出最短的。
定义: dp(u,k)dp(u,k)dp(u,k) 表示有多少种确定 uuu 子树内的边的颜色的方案,使得所有下端点在 uuu 子树内并且尚未满足的条件 的上端点的深度最大值恰好为 kkk。如果所有上述条件都满足, k=0k=0k=0。
人话翻译:
考虑 uuu 子树内能影响到的条件,分为下列两种:
- 上端点在子树内(显然下端点就在子树内了)。如果这种条件没有满足,就永远不可能满足了,这个时候上面的定义表现为 k≥depuk\geq dep_uk≥depu,后面可以看到这部分状态是无用的。
- 上端点是 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=0∑depudp(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)=k∑dp(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=0∑depudp(v,i)+dp(u,k)i=0∑kdp(v,i)+dp(v,k)i=0∑k−1dp(u,i)
长这样子的式子都可以考虑线段树合并。
∑i=0depudp(v,i)\sum_{i=0}^{dep_u}dp(v,i)∑i=0depudp(v,i) 是个常数,先算出来。
合并的时候顺便维护左边遍历过的结点的和,如果一边的结点为空,用维护的和给另一边的结点打乘法标记。递归到叶结点了再处理求和符号的边界情况。
注意维护的这个和是合并前的,要先维护再打标记。可以在递归的时候传引用。
复杂度 O(nlogn+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;
}
【NOI2020】命运【树形dp】【线段树合并】相关推荐
- P6847-[CEOI2019]Magic Tree【dp,线段树合并】
正题 题目链接:https://www.luogu.com.cn/problem/P6847 题目大意 nnn个点的一棵树上,每个时刻可以割掉一些边,一些节点上有果实表示如果在did_idi时刻这个 ...
- Codeforces 671D. Roads in Yusland(树形DP+线段树)
调了半天居然还能是线段树写错了,药丸 这题大概是类似一个树形DP的东西.设$dp[i]$为修完i这棵子树的最小代价,假设当前点为$x$,但是转移的时候我们不知道子节点到底有没有一条越过$x$的路.如果 ...
- P6773-[NOI2020]命运【线段树合并,树形dp】
正题 题目链接:https://www.luogu.com.cn/problem/P6773 题目大意 nnn个点的一棵树,边权可以是000或111.mmm个条件(x,y)(x,y)(x,y)表示要求 ...
- BZOJ5419[Noi2018]情报中心——线段树合并+虚树+树形DP
题目链接: [NOI2018]情报中心 题目大意:给出一棵n个节点的树,边有非负边权,并给出m条链,对于每条链有一个代价,要求选出两条有公共边的链使两条链的并的边权和-两条链的代价和最大. 花了一天的 ...
- 【九省联考2018】秘密袭击【树形dp】【生成函数】【线段树合并】【多项式插值】
题意:nnn 个点的带点权的树,点权最大值为 www,求所有连通子图第 kkk 大权值之和模 641236412364123. n,w≤1666n,w\leq 1666n,w≤1666,时限 5s. ...
- Codeforces.700E.Cool Slogans(后缀自动机 线段树合并 DP)
题目链接 \(Description\) 给定一个字符串\(s[1]\).一个字符串序列\(s[\ ]\)满足\(s[i]\)至少在\(s[i-1]\)中出现过两次(\(i\geq 2\)).求最大的 ...
- 【BZOJ5461】 【PKUWC2018】—Minimax(线段树合并优化dp)
传送门 发现其实就是左右2棵子树,左儿子选到某个值的概率就是 选最大值的概率∗右儿子的值比它小的概率选最大值的概率*右儿子的值比它小的概率选最大值的概率∗右儿子的值比它小的概率 +选最小值的概率∗右儿 ...
- CF700E-Cool Slogans【SAM,线段树合并,dp】
正题 题目链接:https://www.luogu.com.cn/problem/CF700E 题目大意 给出一个字符串SSS,求一个最大的kkk使得存在kkk个字符串其中s1s_1s1是SSS的子 ...
- 线段树合并与分裂维护树上最长上升子序列 + 点分治删点 ---- 2021 牛客多校第一场 C - Cut the tree(详解)
题目大意: 给你一个树,树上每个点都有一个权值valnodeval_{node}valnode,路径(u,v)(u,v)(u,v) 上所有点按顺序有序序列,令f(u,v)f(u,v)f(u,v)是这 ...
最新文章
- FPGA之道(8)时钟处理单元
- stm32 str转hex_【SW4STM32生成 hex文件的设置方法】
- python join()函数
- 清北学堂模拟赛d3t2 b
- openresty获取nginx 请求方法
- Silverlight Telerik控件学习:主题Theme切换
- 大数据之-Hadoop之HDFS_hadoop集群中的安全模式_操作案例---大数据之hadoop工作笔记0075
- nps内网穿透_内网穿透工具:NPS的使用
- RK3128功能介绍
- matlab插值函数截断误差,插值及其误差
- 电脑录屏软件哪个好?高清流畅的录屏方法在这里!
- php 401认证,给phpMyAdmin增加401加密认证
- 干货|建模3D Max中常见问题
- 高德地图初体验demo
- VUE项目中调用高德地图
- 解决flutter应用模拟器出现System UI isn‘t respon报错
- java web统计报表_用Java在Web页面上输出统计图 统计报表
- vs 怎么调试html5,关于html5:使用-VSCode-如何进行远程开发调试
- IDEA插件:多线程文件下载插件开发
- 计算机科学与技术专业适合什么人,计算机科学与技术专业怎么样 主要学什么内容...
热门文章
- 最多金的编程语言Top10:Python第3,R第10,你猜第1是谁?
- 2017新生儿爆款名字出炉!90后的父母们最受欢迎的居然是.....
- python关联分析sklearn_Python3利用pandas,sklearn进行关联度分析以及预测的demo
- 单文件组件的组件传值_移动端组件化架构(下)
- echart 地图 某个地区_中国饮食地图来袭!你爱的口味暗藏健康隐患吗?
- c++歌手大赛系统_计人即讯|第十届程序设计大赛
- java jli.dll_JVM、JRE、JDK之间的区别和联系,你居然还不知道?
- python3抓取图片_通过Python3 爬虫抓取漫画图片
- php表格单元格怎么实现排序,javascript实现对表格元素进行排序操作
- 刷新mac地址命令_配置好Cisco交换机需要熟悉IOS命令及相关的知识