HDU 5029 Relief grain —— 离线树链剖分

This way

题意：

给你一棵树，一开始每个点都是一个空集合，每次在x到y的路径上的所有点的集合中塞进z，问你所有数塞完后每个点中的集合最多的数是什么

题解：

n次插入啊，每次都是一个链，那么我们可以用树链剖分的思想将一条链分为最多log*2的连续区间。那么将所有输入分解成区间后按照dfs序排序，按dfs序做即可。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+5;
struct node
{
    int to,next;
}e[N*2];
int head[N],cnt,tim,fa[N],son[N],dep[N],top[N],siz[N],pos[N];
int mx[N*4];
void add(int x,int y)
{
    e[cnt].to=y;
    e[cnt].next=head[x];
    head[x]=cnt++;
}
void dfs1(int u)
{
    siz[u]=1;
    for (int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if (v!=fa[u])
        {
            fa[v]=u;
            dep[v]=dep[u]+1;
            dfs1(v);
            siz[u]+=siz[v];
            if (son[u]==0||siz[v]>siz[son[u]])
                son[u]=v;
        }
    }
}
int pos_id[N];
void dfs2(int u,int f)
{
    tim++;
    top[u]=f;
    pos[u]=tim;
    pos_id[tim]=u;
    if (son[u]) dfs2(son[u],f);
    for (int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if (v!=fa[u]&&v!=son[u])
        {
            dfs2(v,v);
        }
    }
}
struct Add
{
    int p,v;
    bool operator< (const Add& aa)const
    {
        return p<aa.p;
    }
}ad[N*40];
struct Del
{
    int p,v;
    bool operator< (const Del& aa)const
    {
        return p<aa.p;
    }
}de[N*40];
int num_add,num_del;
void Change(int x,int y,int val)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ad[++num_add]={pos[top[x]],val};
        de[++num_del]={pos[x],val};
        x=fa[top[x]];
    }
    if (pos[x]>pos[y]) swap(x,y);
    ad[++num_add]={pos[x],val};
    de[++num_del]={pos[y],val};
}
int ans[N];

void update(int l,int r,int root,int p,int v)
{
    if(l==r)
    {
        mx[root]+=v;
        return ;
    }
    int mid=l+r>>1;
    if(mid>=p)
        update(l,mid,root<<1,p,v);
    else
        update(mid+1,r,root<<1|1,p,v);
    mx[root]=max(mx[root<<1],mx[root<<1|1]);
}
int query(int l,int r,int root)
{
    if(mx[root]==0)
        return 0;
    if(l==r)
        return l;
    int mid=l+r>>1;
    if(mx[root<<1]==mx[root])
        return query(l,mid,root<<1);
    return query(mid+1,r,root<<1|1);
}
void init(int n)
{
    cnt=num_add=num_del=tim=0;
    for(int i=0;i<=n;i++)
        son[i]=fa[i]=dep[i]=siz[i]=top[i]=pos[i]=0,head[i]=-1;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0)
            break;
        init(n);
        int x,y,w;
        for(int i=1;i<n;i++)
            scanf("%d%d",&x,&y),add(x,y),add(y,x);
        dfs1(1),dfs2(1,1);
        for(int i=1;i<=m;i++)
            scanf("%d%d%d",&x,&y,&w),Change(x,y,w);
        sort(ad+1,ad+1+num_add),sort(de+1,de+1+num_del);
        int p_add=1,p_del=1;
        for(int i=1;i<=n;i++)
        {
            while(p_add<=num_add&&ad[p_add].p==i)
                update(1,1e5,1,ad[p_add].v,1),p_add++;
            ans[pos_id[i]]=query(1,1e5,1);
            while(p_del<=num_del&&de[p_del].p==i)
                update(1,1e5,1,de[p_del].v,-1),p_del++;
        }
        for(int i=1;i<=n;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}