[BZOJ4336][BJOI2015]骑士的旅行(树链剖分+线段树)

树链剖分，对每个叶子用multiset记录前K大士兵，其余节点通过从儿子归并维护前K大士兵。过于模板。

#include<set>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define ls (x<<1)
#define rs (ls|1)
#define lson ls,L,mid
#define rson rs,mid+1,R
#define rep(i,l,r) for (int i=(l); i<=(r); i++)
#define For(i,x) for (int i=h[x],k; i; i=nxt[i])
using namespace std;

const int N=40010;
int n,m,u,v,op,x,y,Q,K,cnt,tim,dfn[N],son[N],sz[N],dep[N],fa[N],top[N],h[N],to[N<<1],nxt[N<<1];
struct P{ int x,k; }p[N];
struct Se{
    int tot,a[30];
    Se(){ tot=0; memset(a,0,sizeof(a)); }
    Se(const Se &b){ tot=b.tot; memcpy(a,b.a,sizeof(a)); }
}T[N<<2];
multiset<int>S[N<<2];
void add(int u,int v){ to[++cnt]=v; nxt[cnt]=h[u]; h[u]=cnt; }

void dfs(int x){
    sz[x]=1; dep[x]=dep[fa[x]]+1;
    For(i,x) if ((k=to[i])!=fa[x]){
        fa[k]=x; dfs(k); sz[x]+=sz[k];
        if (sz[k]>sz[son[x]]) son[x]=k;
    }
}

void dfs2(int x,int tp){
    dfn[x]=++tim; top[x]=tp;
    if (son[x]) dfs2(son[x],tp);
    For(i,x) if ((k=to[i])!=fa[x] && k!=son[x]) dfs2(k,k);
}

Se merge(Se a,Se b){
    Se res;
    for (int i=1,j=1; i<=a.tot || j<=b.tot; ){
        if ((i<=a.tot) && (j>b.tot || b.a[j]<a.a[i])) res.a[++res.tot]=a.a[i++]; else res.a[++res.tot]=b.a[j++];
        if (res.tot==K) break;
    }
    return res;
}

void mdf(int x,int L,int R,int pos,int k,int op){
    if (L==R){
        if (!op) S[x].erase(S[x].find(k)); else S[x].insert(k);
        T[x].tot=0; multiset<int>::iterator it=S[x].end();
        while (1){
            if (it==S[x].begin()) break; it--;
            T[x].a[++T[x].tot]=*it;
            if (T[x].tot==K) break;
        }
        return;
    }
    int mid=(L+R)>>1;
    if (pos<=mid) mdf(lson,pos,k,op); else mdf(rson,pos,k,op);
    T[x]=merge(T[ls],T[rs]);
}

Se que(int x,int L,int R,int l,int r){
    if (L==l && r==R) return T[x];
    int mid=(L+R)>>1;
    if (r<=mid) return que(lson,l,r);
    else if (l>mid) return que(rson,l,r);
        else return merge(que(lson,l,mid),que(rson,mid+1,r));
}

void solve(int x,int y){
    Se res;
    for (; top[x]!=top[y]; x=fa[top[x]]){
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        res=merge(res,que(1,1,n,dfn[top[x]],dfn[x]));
    }
    if (dep[x]<dep[y]) swap(x,y);
    res=merge(res,que(1,1,n,dfn[y],dfn[x]));
    if (!res.tot) puts("-1");
        else{ rep(i,1,res.tot) printf("%d ",res.a[i]); puts(""); }
}

int main(){
    freopen("bzoj4336.in","r",stdin);
    freopen("bzoj4336.out","w",stdout);
    scanf("%d",&n);
    rep(i,2,n) scanf("%d%d",&u,&v),add(u,v),add(v,u);
    dfs(1); dfs2(1,1);
    scanf("%d",&m);
    rep(i,1,m) scanf("%d%d",&p[i].k,&p[i].x);
    scanf("%d%d",&Q,&K);
    rep(i,1,m) mdf(1,1,n,dfn[p[i].x],p[i].k,1);
    while (Q--){
        scanf("%d%d%d",&op,&x,&y);
        if (op==1) solve(x,y);
        if (op==2) mdf(1,1,n,dfn[p[x].x],p[x].k,0),p[x].x=y,mdf(1,1,n,dfn[p[x].x],p[x].k,1);
        if (op==3) mdf(1,1,n,dfn[p[x].x],p[x].k,0),p[x].k=y,mdf(1,1,n,dfn[p[x].x],p[x].k,1);
    }
    return 0;
}