[USACO15DEC] Max Flow P

总时间限制: 

15000ms

单个测试点时间限制: 

1000ms

内存限制: 

524288kB

描述

FJ 给他的 N 个牛棚之间安装了 N−1 根管道，牛棚编号从 1 到 N。所有牛棚都能经由管道相互连通了。

FJ 有 K 条运输牛奶的路线，第 i 条路线从牛棚 si 运输到牛棚 ti。一条运输路线会给它所经过的所有牛棚（包括途中和两个端点处）增加来一个单位的运输压力。每次增加一条路线，你需要计算并输出当前压力最大的牛棚的压力以及这样的牛棚的个数（即压力最大值和最大值的个数）。

输入

第一行输入两个整数 N 和 K。
接下来 N−1 行每行输入两个整数 x 和 y，其中 x≠y。表示一根在牛棚 x 和 y 之间的管道。
接下来 K 行每行两个整数 s 和 t，描述一条从 s 到 t 的运输牛奶的路线。

2<=N<=5*10^4;
1<=K<=10^5。

输出

输出共 K 行，每行两个整数 a,b 分别表示当前最大压力值和这样的牛棚个数，用一个空格隔开。

对于这道题我们有两种做法，先来讲比较简单易懂的树上差分做法，对于这种区间加并且离线询问的题目我们很容易想到差分，但是对于一个父节点有多个儿子，于是我们选择在两个儿子处进行加法，因为在它们从LCA处也需要安装路线，所以我们只在LCA打上-1的标记，然后在LCA的父节点处再打上-1的标记。然后在一系列操作结束后，我们从叶子节点向上跳，当一个节点的所有儿子都访问过后，就将它也加入队列进行访问，相当于是一个拓扑。

代码1：

#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define pii pair<ll,ll>
using namespace std;
const int maxn=1e5+10,L=20;
const ll inf=1e18+7;
int n,rt,m,k;
int d[maxn],fa[maxn][L];
int v[maxn],v2[maxn],mx;
int deg[maxn];
string s;
vector<int> e[maxn]; 
queue<int> q;
inline void dfs(int x,int pa){
	d[x]=d[pa]+1,fa[x][0]=pa;
	for(int i=1;i<=k;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
	for(int y:e[x]) if(y!=pa) deg[x]++,dfs(y,x);
}
inline int lca(int x,int y){
	if(d[y]>d[x]) swap(x,y);
	for(int i=k;~i;i--) if(d[fa[x][i]]>=d[y]) x=fa[x][i];
	if(x==y) return x;
	for(int i=k;~i;i--) if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	//freopen("test.in","r",stdin);
	cin>>n>>m;k=log2(n);
	for(int i=1;i<n;i++){
		int u,v;cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	dfs(1,0);
	while(m--){
		int x,y;cin>>x>>y;
		int l=lca(x,y); 
		v[l]--;
		v[fa[l][0]]--;
		v[x]++,v[y]++;
	} 
	for(int i=2;i<=n;i++) if(!deg[i]) q.push(i);
	while(q.size()){
		int x=q.front();q.pop();
		if(!x) continue;
		mx=max(mx,v[x]);
		v[fa[x][0]]+=v[x];
		if(!--deg[fa[x][0]]) q.push(fa[x][0]);
	}
	cout<<mx<<endl;
	return 0;
}

对于第二种方法，我们采用比较麻烦的树链剖分，当然可维护的信息也就更多了，其实只需要维护最大值和懒标签即可。

代码2：

#include<bits/stdc++.h>
#define ll long long
#define lson (p<<1)
#define rson (p<<1|1)
#define pii pair<int,int>
#define endl '\n'
using namespace std;
const int maxn=1e5+10;
const ll inf=1e18+7;
int n,m,tot;
int mx[maxn<<2],tag[maxn<<2],cnt[maxn<<2];
int d[maxn],sz[maxn],fa[maxn],id[maxn];
int son[maxn],top[maxn];
vector<int> e[maxn];
void dfs(int x,int pa){
	fa[x]=pa,d[x]=d[pa]+1,sz[x]=1;
	for(int y:e[x]){
		if(y==fa[x]) continue;
		dfs(y,x);
		sz[x]+=sz[y];
		if(sz[son[x]]<sz[y]) son[x]=y;
	}
}
void dfs2(int x,int topf){
	id[x]=++tot,top[x]=topf;
	if(!son[x]) return;
	dfs2(son[x],topf);
	for(int y:e[x]) if(y!=fa[x]&&y!=son[x]) dfs2(y,y);
}
inline void pushup(int p){
	mx[p]=max(mx[lson],mx[rson]);
	cnt[p]=0;
	if(mx[p]==mx[lson]) cnt[p]+=cnt[lson];
	if(mx[p]==mx[rson]) cnt[p]+=cnt[rson];
}
inline void pushdown(int p){
	tag[lson]+=tag[p],tag[rson]+=tag[p];
	mx[lson]+=tag[p],mx[rson]+=tag[p];
	tag[p]=0;
}
void build(int p,int l,int r){
	if(l==r){
		cnt[p]=1;
		return;
	}
	int mid=l+r>>1;
	build(lson,l,mid);
	build(rson,mid+1,r);
	pushup(p);
}
void change(int p,int l,int r,int L,int R,int v){
	if(l>r) return;
	if(L<=l&&r<=R){
		mx[p]+=v;
		tag[p]+=v;
		return;
	}
	pushdown(p);
	int mid=l+r>>1;
	if(mid>=L) change(lson,l,mid,L,R,v);
	if(R>=mid+1) change(rson,mid+1,r,L,R,v);
	pushup(p);
}
void changet(int x,int y,int v){
	while(top[x]!=top[y]){
		if(d[top[x]]<d[top[y]]) swap(x,y);
		change(1,1,n,id[top[x]],id[x],v);
		x=fa[top[x]];
	}
	if(d[x]>d[y]) swap(x,y);
	change(1,1,n,id[x],id[y],v);
	//printf("x:%d idx:%d\n",x,id[x]);
	//printf("y:%d idy:%d\n",y,id[y]);
}
int main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	//freopen("test.in","r",stdin);
	cin>>n>>m;
	for(int i=1;i<n;i++){
		int u,v;cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	dfs(1,0);
	dfs2(1,1);
	build(1,1,n);
	while(m--){
		int x,y;cin>>x>>y;
		changet(x,y,1);
		//cout<<mx[1]<<' '<<cnt[1]<<endl;
	}
	cout<<mx[1]<<endl;
	return 0;
}