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 longusing 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;}