传送门
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\n",ans[i]);
return 0;
}
Blog来自PaperCloud,未经允许,请勿转载,TKS!