[洛谷 P4556] 雨天的尾巴

传送门

Solution

线段树合并的入门题
lca可以在dfs的时候离线求（用并查集）
更新的点有每条链的两个端点，它们的lca和dad[lca]
为了节省空间，lca和dad[lca]的更新可以先存储下来，等搜完该点后再做

##Code

#include<bits/stdc++.h>
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
	return x*f;
}
#define MN 100005
#define N 100000 
int n,m,z[MN],ans[MN];
struct ques{int b,nex;}q[MN<<1];
int qhr[MN],cnt=1;
struct edge{int to,nex;}e[MN<<1];
int hr[MN],en;
inline void qins(int f,int t)
{
	q[++cnt]=(ques){t,qhr[f]};qhr[f]=cnt;
	if(f==t) ++cnt;
	else q[++cnt]=(ques){f,qhr[t]},qhr[t]=cnt;
}
inline void ins(int f,int t)
{
	e[++en]=(edge){t,hr[f]};hr[f]=en;
	e[++en]=(edge){f,hr[t]};hr[t]=en;
} 
int fa[MN],dad[MN];
bool vis[MN];
inline int getf(int x){return fa[x]==x?x:fa[x]=getf(fa[x]);}
#define mid ((l+r)>>1) 
int ls[MN*40],rs[MN*40],rt[MN],t[MN*40],seg;
void merge(int &x,int y,int l,int r)
{
	if(!x||!y){x=x+y;return;}	
	if(l==r){t[x]+=t[y];return;}
	merge(ls[x],ls[y],l,mid);
	merge(rs[x],rs[y],mid+1,r);
	t[x]=max(t[ls[x]],t[rs[x]]);
}
int query(int x,int l,int r)
{
	if(!x||!t[x]) return 0;
	if(l==r) return l;
	return t[rs[x]]>t[ls[x]]?query(rs[x],mid+1,r):query(ls[x],l,mid);
}
void update(int &x,int l,int r,int p,int v)
{
	if(!x) x=++seg;
	if(l==r){t[x]+=v;return;}
	if(p>mid) update(rs[x],mid+1,r,p,v);
	if(p<=mid) update(ls[x],l,mid,p,v);
	t[x]=max(t[ls[x]],t[rs[x]]);
}
std::map<int,int> mar[MN];
void dfs(int x,int f)
{
	register int i;
	fa[x]=x;dad[x]=f;
	for(i=hr[x];i;i=e[i].nex)if(f^e[i].to)
	{
		dfs(e[i].to,x);
		fa[e[i].to]=x;
		merge(rt[x],rt[e[i].to],1,N);
	}
	vis[x]=true;
	for(i=qhr[x];i;i=q[i].nex)if(vis[q[i].b]==true)
	{
		int lca=getf(q[i].b);
		mar[lca][z[i>>1]]--;mar[dad[lca]][z[i>>1]]--;
	}
	std::map<int,int>::iterator it;
	for(it=mar[x].begin();it!=mar[x].end();++it) update(rt[x],1,N,it->first,it->second);
	ans[x]=query(rt[x],1,N);
}
int main()
{
	n=read(),m=read();
	register int i,j,k;
	for(i=1;i<n;++i) ins(read(),read());
	for(i=1;i<=m;++i)
	{
		j=read(),k=read(),z[i]=read();
		qins(j,k),update(rt[j],1,N,z[i],1),update(rt[k],1,N,z[i],1);
	}
	dfs(1,0);
	for(i=1;i<=n;++i) printf("%d
",ans[i]);
	return 0;
}

---

Blog来自PaperCloud，未经允许，请勿转载，TKS！