bzoj3307 雨天的尾巴（树上差分+线段树合并）

每个点一颗权值线段树，因为只有最后一次询问，因此我们可以树上差分，最后一遍dfs合并出来。点数应该最多是 4n 4 n 的，所以复杂度应该还是 O((n+m)log1e9) O ( ( n + m ) l o g 1 e 9 ) 的

#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(T==S){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=0,fa[N][20],Log[N],dep[N],rt[N],owo=0,ans[N];
struct edge{
    int to,next;
}data[N<<1];
struct Data{
    int cnt,id;
    friend bool operator<(Data x,Data y){
        return x.cnt==y.cnt?x.id>y.id:x.cntvoid operator+=(Data x){
        cnt+=x.cnt;
    }
};
struct node{
    int lc,rc;Data x;
}tr[N*60];
inline void dfs(int x){
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x][0]) continue;
        fa[y][0]=x;dep[y]=dep[x]+1;dfs(y);
    }
}
inline int lca(int x,int y){
    if(dep[x]int d=dep[x]-dep[y];
    for(int i=0;i<=Log[d];++i)
        if(d>>i&1) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
inline void pushup(int p){
    tr[p].x=max(tr[tr[p].lc].x,tr[tr[p].rc].x);
}
inline void ins(int &p,int l,int r,int x,int val){
    if(!p) p=++owo;if(l==r){tr[p].x.cnt+=val;tr[p].x.id=l;return;}
    int mid=l+r>>1;
    if(x<=mid) ins(tr[p].lc,l,mid,x,val);
    else ins(tr[p].rc,mid+1,r,x,val);pushup(p);
}
inline void merge(int &p1,int p2,int l,int r){
    if(!p1||!p2){p1=p1+p2;return;}
    if(l==r){tr[p1].x+=tr[p2].x;return;}int mid=l+r>>1;
    merge(tr[p1].lc,tr[p2].lc,l,mid);
    merge(tr[p1].rc,tr[p2].rc,mid+1,r);pushup(p1);
}
inline void dfs1(int x){
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x][0]) continue;
        dfs1(y);merge(rt[x],rt[y],1,1e9);
    }ans[x]=tr[rt[x]].x.id;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();Log[0]=-1;
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;iint x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1);
    while(m--){
        int x=read(),y=read(),w=read(),t=lca(x,y);
        ins(rt[x],1,1e9,w,1);ins(rt[y],1,1e9,w,1);
        ins(rt[t],1,1e9,w,-1);if(fa[t][0]) ins(rt[fa[t][0]],1,1e9,w,-1);
    }dfs1(1);
    for(int i=1;i<=n;++i) printf("%d\n",ans[i]);
    return 0;
}