BZOJ3307 雨天的尾巴

Problem

BZOJ权限题
良心洛谷

Solution

这道题目的花呢，本来想缩先树剖(可能需要长链剖分?)，再搞一个标记永久化，然后最后合并答案，不过可能会卡空间。。时间复杂度呢是$O(nlog^2n)$
然后boshi表示：“为什么不差分然后再自底向上合并线段树呢”
于是就做到了$O(nlogn)$
好优秀的做法qwqqq
窝getlca写残了导致调了很久

Code

#include <algorithm>
#include <cstdio>
#define pushup(x) mx[x]=max(mx[lc[x]],mx[rc[x]])
using namespace std;
const int maxn=100010,maxm=5000010;
struct data{int v,nxt;}edge[maxn<<1];
int n,m,p,tot,head[maxn],deep[maxn],f[maxn][21],ans[maxn];
int rt[maxn],mx[maxm],lc[maxm],rc[maxm];
template <typename Tp> inline void read(Tp &x)
{
    x=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
inline int max(int x,int y){return x>y?x:y;}
inline void insert(int u,int v)
{
    edge[++p]=(data){v,head[u]};head[u]=p;
    edge[++p]=(data){u,head[v]};head[v]=p;
}
void input()
{
    int u,v;
    read(n);read(m);
    for(int i=1;i<n;i++)
    {
        read(u);read(v);
        insert(u,v);
    }
}
void dfs1(int x)
{
    for(int i=1;i<21;i++) f[x][i]=f[f[x][i-1]][i-1];
    for(int i=head[x];i;i=edge[i].nxt)
      if(edge[i].v!=f[x][0])
        deep[edge[i].v]=deep[x]+1,f[edge[i].v][0]=x,dfs1(edge[i].v);
}
int getlca(int x,int y)
{
    if(deep[x]<deep[y]) swap(x,y);
    int t=deep[x]-deep[y];
    for(int i=0;i<21;i++)
      if(t&(1<<i))
        x=f[x][i];
    if(x==y) return x;
    for(int i=20;~i;i--)
      if(f[x][i]!=f[y][i])
        x=f[x][i],y=f[y][i];
    return f[x][0];
}
void update(int l,int r,int pos,int val,int &rt)
{
    if(!rt) rt=++tot;
    if(l==r){mx[rt]+=val;return ;}
    int m=(l+r)>>1;
    if(pos<=m) update(l,m,pos,val,lc[rt]);
    else update(m+1,r,pos,val,rc[rt]);
    pushup(rt);
}
int query(int l,int r,int rt)
{
    if(l==r) return l;
    int m=(l+r)>>1;
    if(mx[rt]==mx[lc[rt]]) return query(l,m,lc[rt]);
    else return query(m+1,r,rc[rt]); 
}
int merge(int l,int r,int x,int y)
{
    if(!x||!y) return x+y;
    if(l==r) {mx[x]+=mx[y];return x;}
    int m=(l+r)>>1;
    lc[x]=merge(l,m,lc[x],lc[y]);
    rc[x]=merge(m+1,r,rc[x],rc[y]);
    pushup(x);
    return x;
}
void dfs2(int x)
{
    for(int i=head[x];i;i=edge[i].nxt)
      if(edge[i].v!=f[x][0])
      {
        dfs2(edge[i].v);
        rt[x]=merge(1,100000,rt[x],rt[edge[i].v]);
      }
    if(mx[rt[x]]==0) ans[x]=0;
    else ans[x]=query(1,100000,rt[x]);
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    int x,y,z,lca;
    input();
    dfs1(1);
    while(m--)
    {
        read(x);read(y);read(z);
        lca=getlca(x,y);
        update(1,100000,z,1,rt[x]);
        update(1,100000,z,1,rt[y]);
        update(1,100000,z,-1,rt[lca]);
        if(lca^1) update(1,100000,z,-1,rt[f[lca][0]]);
    }
    dfs2(1);
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
    return 0;
}