[洛谷 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\n",ans[i]);
    return 0;
}

---

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

转载于:https://www.cnblogs.com/PaperCloud/p/10087530.html