BZOJ_3307_雨天的尾巴_线段树合并+树上差分

BZOJ_3307_雨天的尾巴_线段树合并

Description

N个点，形成一个树状结构。有M次发放，每次选择两个点x,y
对于x到y的路径上（含x,y)每个点发一袋Z类型的物品。完成
所有发放后，每个点存放最多的是哪种物品。

Input

第一行数字N，M
接下来N-1行，每行两个数字a,b,表示a与b间有一条边
再接下来M行，每行三个数字x,y,z.如题

Output

输出有N行
每i行的数字表示第i个点存放最多的物品是哪一种，如果有
多种物品的数量一样，输出编号最小的。如果某个点没有物品
则输出0

 

---

 

类似天天爱跑步，树上差分，在每个点的线段树上修改，然后自下向上合并上去。

然后线段树怎么搞。。维护全局众数，需要同时维护众数是谁以及众数出现了多少次。

注意由于我们把叶子结点当作这个数出现的次数，线段树合并到底时需要加上这个数的个数。

对于无解的情况要判掉，即众数出现的次数为0，因为可能左右的众数都不为0但出现次数为0。

然后这题我迷之TLE，然后把权值离散化才卡过的？？？

 

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 100050
#define maxn m
int t[N*60],n,m,head[N],to[N<<1],nxt[N<<1],cnt,tot,mx[N*60],xx[N],yy[N],zz[N];
int fa[N],top[N],son[N],siz[N],dep[N],root[N],ls[N*60],rs[N*60],ans[N],v[N];
inline void add(int u,int v) {
	to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt;
}
void dfs1(int x,int y) {
	int i; siz[x]=1; fa[x]=y; dep[x]=dep[y]+1;
	for(i=head[x];i;i=nxt[i]) {
		if(to[i]!=y) {
			dfs1(to[i],x);
			siz[x]+=siz[to[i]];
			if(siz[to[i]]>siz[son[x]]) son[x]=to[i];
		}
	}
}
void dfs2(int x,int t) {
	top[x]=t; 
	if(son[x]) dfs2(son[x],t);
	int i;
	for(i=head[x];i;i=nxt[i]) {
		if(to[i]!=son[x]&&to[i]!=fa[x]) {
			dfs2(to[i],to[i]);
		}
	}
}
int lca(int x,int y) {
	while(top[x]!=top[y]) {
		if(dep[top[x]]>dep[top[y]]) swap(x,y);
		y=fa[top[y]];
	}
	return dep[x]<dep[y]?x:y;
}
void pushup(int p) {
	if(t[ls[p]]>=t[rs[p]]) mx[p]=mx[ls[p]],t[p]=t[ls[p]];
	else mx[p]=mx[rs[p]],t[p]=t[rs[p]];
}
int merge(int l,int r,int x,int y) {
	if(!x) return y;
	if(!y) return x;
	if(l==r) {
		t[x]+=t[y]; return x;
	}
	int mid=(l+r)>>1;
	ls[x]=merge(l,mid,ls[x],ls[y]);
	rs[x]=merge(mid+1,r,rs[x],rs[y]);
	pushup(x);
	return x;	
}
void update(int l,int r,int x,int v,int &p) {
	if(!p) p=++tot;
	if(l==r) {
		t[p]+=v; mx[p]=l; return ;
	}
	int mid=(l+r)>>1;
	if(x<=mid) update(l,mid,x,v,ls[p]);
	else update(mid+1,r,x,v,rs[p]);
	pushup(p);
}
void dfs3(int x) {
	int i;
	for(i=head[x];i;i=nxt[i]) {
		if(to[i]!=fa[x]) {
			dfs3(to[i]);
			root[x]=merge(1,maxn,root[x],root[to[i]]);
		}
	}
	if(t[root[x]]) ans[x]=v[mx[root[x]]];
}
int main() {
	scanf("%d%d",&n,&m);
	int i,x,y,z;
	for(i=1;i<n;i++) {
		scanf("%d%d",&x,&y); add(x,y); add(y,x);
	}
	dfs1(1,0);
	dfs2(1,1);
	for(i=1;i<=m;i++) {
		scanf("%d%d%d",&xx[i],&yy[i],&zz[i]);
		v[i]=zz[i];
	}
	sort(v+1,v+m+1);
	for(i=1;i<=m;i++) {
		zz[i]=lower_bound(v+1,v+m+1,zz[i])-v,x=xx[i],y=yy[i];
		int l=lca(x,y);
		update(1,maxn,zz[i],1,root[x]);
		update(1,maxn,zz[i],1,root[y]);
		update(1,maxn,zz[i],-1,root[l]);
		if(fa[l]) update(1,maxn,zz[i],-1,root[fa[l]]);
	}
	dfs3(1);
	for(i=1;i<=n;i++) {
		printf("%d\n",ans[i]);
	}
}
/*
1<=N,M<=100000
1<=a,b,x,y<=N
1<=z<=10^9
*/