你才会停止追逐远方|

ricky_lin

园龄:3年9个月粉丝:11关注:2

P6773 [NOI2020] 命运

标签:DS B | DP B

题意 给定一棵 nm$ 条祖先到儿子的路径,求给边黑白染色的方案数使得每条路径上至少有一个黑点。

n,m5×105

你考虑显然如果一个子树内有路径没有被消除,并且延伸到子树外,那么我们显然只关心深度最深的那个点。

那么我们设 fu,j 表示子树 u 中没有被消除过的路径中祖先节点最深的深度为 j 时的方案数。

我们枚举一条边是否染成黑色,可以得到转移如下:

fu,i=j=0depufu,ifv,j+j=0ifu,ifv,j+j=0i1fu,jfv,i

显然我们可以将与 j 无关的项提出来,就变成了下式:

fu,i=fu,i(j=0depufv,j+j=0ifv,j)+fv,ij=0i1fu,j

我们设 sumi,j=k=0jfi,k 那么就可以得到一个 n2 的转移:

fu,i=fu,i(sumv,depu+sumv,i)+fv,isumu,i1

我们考虑因为只有 m 个询问,所以说 dp 时很多状态和转移都是没有什么必要的,但是我们也不能直接优化掉转移。

考虑对于这种类型的 dp 我们常见的就是用 set+启发式合并线段树合并长链剖分 等方式进行优化。

显然第一种是 O(nlog2n) 的,不可接受,所以我们使用 线段树合并 解决这个问题。

考虑 sumv,depu 可以提前求出,而 sumv,i,sumu,i1 可以在线段树的过程中动态增加,至于和对应权值的乘法就直接打乘法标记即可,合并的时候会将两个部分的答案加起来。

code:

#include<bits/stdc++.h>
using namespace std;
const int NN = 5e5 + 8,MOD = 998244353;
typedef long long ll;
int n,m;
inline int read(){
register char c = getchar();
register int res = 0;
while(!isdigit(c)) c = getchar();
while(isdigit(c)) res = res * 10 + c - '0', c = getchar();
return res;
}
struct Edge{
int to,next;
}edge[NN << 1];
int head[NN],cnt;
void init(){
memset(head,-1,sizeof(head));
cnt = 1;
}
void add_edge(int u,int v){
edge[++cnt] = {v,head[u]};
head[u] = cnt;
}
struct Seg{
int ls,rs;
ll num,mul;
#define ls(x) tree[x].ls
#define rs(x) tree[x].rs
#define num(x) tree[x].num
#define mul(x) tree[x].mul
}tree[NN << 5];
int nodecnt;
void addlz(int x,ll num){
if(!x) return;
num(x) = num(x) * num % MOD;
mul(x) = mul(x) * num % MOD;
}
void pushup(int x){
num(x) = (num(ls(x)) + num(rs(x))) % MOD;
}
void pushdown(int x){
addlz(ls(x),mul(x));
addlz(rs(x),mul(x));
mul(x) = 1;
}
void build(int &x,int l,int r,int pos){
if(!x) x = ++nodecnt;
num(x) = mul(x) = 1;
if(l == r) return;
int mid = (l + r) / 2;
if(pos <= mid) build(ls(x),l,mid,pos);
else build(rs(x),mid + 1,r,pos);
}
ll query(int x,int l,int r,int pos){
if(!x || r <= pos) return num(x);
int mid = (l + r) / 2;
ll res = 0;
pushdown(x);
if(pos <= mid) return query(ls(x),l,mid,pos);
else return (num(ls(x)) + query(rs(x),mid+1,r,pos)) % MOD;
}
// s1 -> (sum[y][dep[x]]+sum[y][i]), s2 -> sum[x][i-1]
int merge(int x,int y,int l,int r,ll &s1,ll &s2){
if(!x && !y) return 0;
if(!x || !y){
if(y){
s1 = (s1 + num(y)) % MOD;
addlz(y,s2);
return y;
}
s2 = (s2 + num(x)) % MOD;
addlz(x,s1);
return x;
}
if(l == r){
ll tx = num(x), ty = num(y);
s1 = (s1 + ty) % MOD;
num(x) = (num(x) * s1 + num(y) * s2) % MOD;
s2 = (s2 + tx) % MOD;
return x;
}
pushdown(x),pushdown(y);
int mid = (l + r) / 2;
ls(x) = merge(ls(x),ls(y),l,mid,s1,s2);
rs(x) = merge(rs(x),rs(y),mid+1,r,s1,s2);
pushup(x);
return x;
}
vector<int> Q[NN];
int dep[NN];
int rt[NN];
void dfs(int u,int fa){
dep[u] = dep[fa] + 1;
int mx = 0;
for(int i : Q[u]) mx = max(mx,dep[i]);
build(rt[u],0,n,mx);
for(int i = head[u]; i != -1; i = edge[i].next){
int v = edge[i].to;
if(v == fa) continue;
dfs(v,u);
ll S1 = query(rt[v],0,n,dep[u]),S2 = 0;
// printf("%lld\n",S1);
rt[u] = merge(rt[u],rt[v],0,n,S1,S2);
}
}
int main(){
n = read();
init();
for(int i = 1,u,v; i < n; ++i){
u = read();v = read();
add_edge(u,v);add_edge(v,u);
}
m = read();
for(int i = 1,u,v; i <= m; ++i){
u = read();v = read();
Q[v].push_back(u);
}
dfs(1,0);
printf("%lld",query(rt[1],0,n,0));
}
posted @   ricky_lin  阅读(4)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示
评论
收藏
关注
推荐
深色
回顶
收起
  1. 1 有我 周深
有我 - 周深
00:00 / 00:00
An audio error has occurred.

作词 : 唐恬/闫光宇

作曲 : 钱雷

编曲 : 赵兆/付虹宇

制作人 : 赵兆

出品 : 共青团中央宣传部

版权 : 中国青少年新媒体协会

制作单位 : 能量悦动音乐

发行单位 : 银河方舟StarNation

出品人 : 郭峰

总监制 : 汤杰

总策划 : 钟亚楠

总统筹 : 金慧子

音乐监制 : 李天鹏/李三木

制作执行 : 张不贰/高聪怡

项目宣发 : 肖健/张国党/孙小千/戴胤/孙雯璟

音乐推广 : 代诗琪/杜思潮/马越/程铁峰/傅之豪

钢琴 : 赵兆

吉他 : 伍凌枫

贝斯 : 韩阳

鼓 : 武勇恒

合唱设计 : 赵兆

合唱 : 凡尔赛合唱团

人声录音 : 耿潇微

人声录音室 : 55TEC Studio Beijing

配唱 : 徐威@52Hz Studio (Shanghai)

混音 : 李游(小骷髅)@55TEC Studio Beijing

海报 : 格子

特别鸣谢 : 周深工作室

世界问 你是谁 来自哪 请回答

爱什么 梦什么 去何方 请回答

答案有 一百年的时光

我来自 硝烟中 课桌旁 的太阳

我来自 硝烟中 课桌旁 的太阳

他和她 宣的誓 迎的仗

来自那 燃烧的 和我一样 的年华

来自世间 一对平凡的夫妻 身旁

来自世间 一对平凡的夫妻 身旁

来自昨天 谁以青春赴万丈 理想

我是寸土 不让的 家乡啊

我是绝不 低头的 倔强啊

接过万千热血 的初衷

当有对答世界 的音量

要怎么形容明天 像我一样

要怎么形容明天 像我一样

承风骨亦有锋芒 有梦则刚

去何方 去最高 的想象

前往皓月星辰 初心不忘

那未来如何登场 有我担当

那未来如何登场 有我担当

定是你只能叫好 那种辉光

护身旁 战远方 有我啊

我的名字就是 站立的地方

Wu~

我的样子 就是 明天的模样

我是朝阳 落在乡间听书声 朗朗

我是朝阳 落在乡间听书声 朗朗

我是屏障 为谁挡一程厄运 的墙

我要一生 清澈地 爱着啊

我要长歌 领着风 踏着浪

朝着星辰大海 的方向

当有对答世界 的音量

要怎么形容明天 像我一样

要怎么形容明天 像我一样

承风骨亦有锋芒 有梦则刚

去远方 去最高 的想象

前往皓月星辰 初心不忘

那未来如何登场 有我担当

那未来如何登场 有我担当

定是你只能叫好 那种辉光

护身旁 战远方 有我啊

一生骄傲为我 站立的地方

Wu~

我的样子 就是 中国的模样

Wu~~~ Wu~~~

当炬火 去化作那道光

“谨以此歌献给一代代不负时代重托的中国青年”

“谨以此歌献给一代代不负时代重托的中国青年”