BZOJ3307: 雨天的尾巴

线段树合并
在链的两端x,y各打上1个z的加标记，在LCA(x,y)打上1个z的减标记，在fa(LCA(x,y))打上1个z的减标记，每个节点和所有孩子的线段树合并，查询时线段树上二分

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

inline void read(int &x)
{
    char c;
    while(!((c=getchar())>='0'&&c<='9'));
    x=c-'0';
    while((c=getchar())>='0'&&c<='9') x=(x<<3)+(x<<1)+c-'0';
}
inline void up(int &x,const int &y){if(x<y)x=y;}
inline void swap(int &x,int &y){x^=y;y^=x;x^=y;}
const int maxn = 100005;
const int maxd = 20;

int n,m;

struct edge
{
    int y,c,nex;
    edge(){}
    edge(const int _y,const int _nex){y=_y;nex=_nex;}
}a[maxn<<1],M[maxn<<2]; int len,fir[maxn],fi[maxn],lenm;
inline void ins(const int x,const int y){a[++len]=edge(y,fir[x]);fir[x]=len;}
inline void ins2(const int x,const int y,const int c){M[++lenm]=edge(y,fi[x]);M[lenm].c=c;fi[x]=lenm;}

int fa[maxn][maxd],dep[maxn];
void dfs(const int x)
{
    for(int i=1;i<maxd;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int k=fir[x];k;k=a[k].nex)
    {
        const int y=a[k].y;
        if(y!=fa[x][0]) dep[y]=dep[x]+1,fa[y][0]=x,dfs(y);
    }
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=maxd-1;i>=0;i--)
        if(dep[x]-dep[y]>=(1<<i)) x=fa[x][i];
    if(x==y) return x;
    for(int i=maxd-1;i>=0;i--)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}

struct node
{
    int x,y,c,v;
}q[maxn];
inline bool cmp(const node x,const node y){return x.c<y.c;}
int LSH[maxn];

struct segment
{
    int lc,rc,c;
}seg[maxn*70]; int tot;
int root[maxn];

inline void pushup(const int x)
{
    int lc=seg[x].lc,rc=seg[x].rc;
    seg[x].c=seg[lc].c>seg[rc].c?seg[lc].c:seg[rc].c;
}
void merge(int &x,const int y,const int l,const int r)
{
    if(!y) return;
    if(!x) { x=y; return; }
    if(l==r) { seg[x].c+=seg[y].c; return; }

    int mid=l+r>>1;
    merge(seg[x].lc,seg[y].lc,l,mid); merge(seg[x].rc,seg[y].rc,mid+1,r);
    pushup(x);
}
int loc,c;
void upd(int &x,const int l,const int r)
{
    if(!x) x=++tot;
    if(l==r) {seg[x].c+=c; return;}
    int mid=l+r>>1;
    if(loc<=mid) upd(seg[x].lc,l,mid);
    else upd(seg[x].rc,mid+1,r);
    pushup(x);
}
int query(const int x,const int l,const int r)
{
    if(l==r)
    {
        if(seg[x].c==0) return 0;
        return l;
    }
    int mid=l+r>>1;
    if(seg[seg[x].lc].c>=seg[seg[x].rc].c) return query(seg[x].lc,l,mid);
    else return query(seg[x].rc,mid+1,r);
}

int maxz;
int ans[maxn];
void solve(const int x)
{
    for(int k=fir[x];k;k=a[k].nex) if(a[k].y!=fa[x][0])
    {
        solve(a[k].y);
        merge(root[x],root[a[k].y],1,maxz);
    }
    for(int k=fi[x];k;k=M[k].nex)
        loc=M[k].y,c=M[k].c,upd(root[x],1,maxz);
    ans[x]=LSH[query(root[x],1,maxz)];
}

int main()
{
    read(n); read(m);
    for(int i=1;i<n;i++)
    {
        int x,y; read(x); read(y);
        ins(x,y); ins(y,x);
    }
    dep[1]=1; dfs(1);

    for(int i=1;i<=m;i++)
    {
        int x,y,v; read(x); read(y); read(v);
        q[i].x=x; q[i].y=y; q[i].c=v;
    }
    sort(q+1,q+m+1,cmp);

    q[0].c=q[1].c-1; maxz=0;
    for(int i=1;i<=m;i++)
    {
        if(q[i].c!=q[i-1].c) LSH[++maxz]=q[i].c;
        q[i].v=maxz;
    }

    for(int i=1;i<=m;i++)
    {
        const int x=q[i].x,y=q[i].y;
        int ff=LCA(x,y);
        ins2(x,q[i].v,1); ins2(y,q[i].v,1);
        ins2(ff,q[i].v,-1); ins2(fa[ff][0],q[i].v,-1);
    }

    solve(1);
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);

    return 0;
}