刷题记录:牛客NC24019Max Flow

传送门:牛客

题目描述

FJ 给他的牛棚的 $N$ 个隔间之间安装了 $N-1$ 根管道，隔间编号从 $1$ 到 $N$。所有隔间都被管道连通了。
FJ 有 $K$ 条运输牛奶的路线，第 $i$ 条路线从隔间 $s_i$ 运输到隔间 $t_i$。一条运输路线会给它的两个端点处的隔间以及中间途径的所有隔间带来一个单位的运输压力，你需要计算压力最大的隔间的压力是多少。

输入:
5 10
3 4
1 5
4 2
5 4
5 4
5 4
3 5
4 3
4 3
1 3
3 5
5 4
1 5
3 4
输出:
9

一道树链剖分+线段树维护区间加+线段树维护区间$max$的题目

首先读完题目发现是在树上进行操作,很显然可以使用树链剖分将树形结构分解为线性结构再进行维护

转化为线性结构后,对于每一次增加的压力,我们可以使用线段树来对这个路线上的所有点都加一,那么最终的答案肯定就是所有点的压力的最大值,也就是区间的$max$.

操作想法并不艰难,只是代码量较大需要注意一些细节

下面是具体的代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int fa[maxn],max_son[maxn],Size[maxn],dep[maxn];
vector<int>edge[maxn];
void dfs1(int u,int per_u) {
	Size[u]=1;
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v==per_u) continue;
		dep[v]=dep[u]+1;
		dfs1(v,u);
		fa[v]=u;Size[u]+=Size[v];
		if(Size[v]>Size[max_son[u]]) {
			max_son[u]=v;
		}
	}
}
int top[maxn],id[maxn],tot=0,rev[maxn];
void dfs2(int u,int t) {
	top[u]=t;id[u]=++tot;
	if(!max_son[u]) return ;
	dfs2(max_son[u],t);
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v==fa[u]||v==max_son[u]) continue;
		dfs2(v,v);
	}
}
struct Segment_tree{
	int l,r,mx,lazy;
}tree[maxn*4];
void pushup(int rt) {
	tree[rt].mx=max(tree[ls].mx,tree[rs].mx);
}
void build(int l,int r,int rt) {
	tree[rt].l=l;tree[rt].r=r;
	if(l==r) {
		return ;
	}
	int mid=(l+r)>>1;
	build(lson);build(rson);
}
void change(int rt,int val) {
	tree[rt].mx+=val;tree[rt].lazy+=val;
}
void pushdown(int rt) {
	change(ls,tree[rt].lazy);change(rs,tree[rt].lazy);
	tree[rt].lazy=0;
}
void update(int l,int r,int rt,int val) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		change(rt,val);
		return ;	
	}
	if(tree[rt].lazy) pushdown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) update(l,r,ls,val);
	else if(l>mid) update(l,r,rs,val);
	else update(l,mid,ls,val),update(mid+1,r,rs,val);
	pushup(rt);
}
int ans_max=-int_INF;
void query(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		ans_max=max(ans_max,tree[rt].mx);
		return ;
	}	
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) query(l,r,ls);
	else if(l>mid) query(l,r,rs);
	else query(l,mid,ls),query(mid+1,r,rs);
}
void Add(int u,int v) {
	while(top[u]!=top[v]) {
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		update(id[top[u]],id[u],1,1);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	update(id[u],id[v],1,1);
}
void ask(int u,int v) {
	while(top[u]!=top[v]) {
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		query(id[top[u]],id[u],1);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	query(id[u],id[v],1);
}
int n,k;
int main() {
	n=read();k=read();
	for(int i=1;i<=n-1;i++) {
		int u=read(),v=read();
		edge[u].push_back(v);
		edge[v].push_back(u);
	}
	dfs1(1,0);dfs2(1,1);
	build(1,tot,1);
	for(int i=1;i<=k;i++) {
		int s=read(),t=read();
		Add(s,t);
	}
	int ans=-int_INF;
	for(int i=2;i<=n;i++) {
		ans_max=-int_INF;
		ask(1,i);
		ans=max(ans,ans_max);
	}
	cout<<ans<<endl;
	return 0;
}