bzoj4336 骑士的旅行 (树链剖分+multiset)

首先大概有一个树剖+树套树的做法，但我哪会写啊

然后发现k很小，如果用线段树记每个区间前k大的的话，可以O(k)地合并

而且一个点还有可能有好多个骑士，所以要用multiset维护一下

然后树剖就好啦

 #include<bits/stdc++.h>
 #define pa pair<int,int>
 #define CLR(a,x) memset(a,x,sizeof(a))
 #define IT multiset<int>::iterator
 using namespace std;
 typedef long long ll;
 const int maxn=4e4+,maxk=;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 int eg[maxn*][],egh[maxn],ect,smr[maxn][];
 int N,M,Q,K;
 int wson[maxn],dfn[maxn],dep[maxn],siz[maxn],top[maxn],fa[maxn],tot;
 int fr[maxn*][maxk],id[maxn],ans[maxk],tmp[maxk];
 multiset<int,greater<int> > st[maxn];

 inline void adeg(int a,int b){
     eg[++ect][]=b;eg[ect][]=egh[a];egh[a]=ect;
 }

 void dfs1(int x){
     siz[x]=;int mm=;
     for(int i=egh[x];i;i=eg[i][]){
         int b=eg[i][];
         if(b==fa[x]) continue;
         fa[b]=x;dep[b]=dep[x]+;
         dfs1(b);siz[x]+=siz[b];
         if(siz[b]>mm) mm=siz[b],wson[x]=b;
     }
 }
 void dfs2(int x){
     dfn[x]=++tot;id[tot]=x;
     top[x]=(x==wson[fa[x]]?top[fa[x]]:x);
     if(wson[x]) dfs2(wson[x]);
     for(int i=egh[x];i;i=eg[i][]){
         int b=eg[i][];
         if(b==wson[x]||b==fa[x]) continue;
         dfs2(b);
     }
 }

 inline void update(int p){
     int l=p<<,r=p<<|;
     int a=,b=,i=;
     for(;a<=fr[l][]&&i<K;a++){
         for(;b<=fr[r][]&&fr[r][b]>fr[l][a]&&i<K;b++)
             fr[p][++i]=fr[r][b];
         if(i<K) fr[p][++i]=fr[l][a];
     }for(;b<=fr[r][]&&i<K;b++) fr[p][++i]=fr[r][b];
     fr[p][]=i;
 }

 void build(int p,int l,int r){
     if(l==r){
         int i=;
         for(IT it=st[id[l]].begin();i<K&&it!=st[id[l]].end();it++){
             fr[p][++i]=*it;
         }fr[p][]=i;
     }else{
         int m=l+r>>;
         build(p<<,l,m);build(p<<|,m+,r);
         update(p);
     }
 }

 void query(int p,int l,int r,int x,int y){
     if(x<=l&&r<=y){
         if(!ans[]){
             memcpy(ans,fr[p],sizeof(ans));
             return;
         }
         int a=,b=,i=;
         for(;a<=fr[p][]&&i<K;a++){
             for(;b<=ans[]&&ans[b]>fr[p][a]&&i<K;b++)
                 tmp[++i]=ans[b];
             if(i<K) tmp[++i]=fr[p][a];
         }for(;b<=ans[]&&i<K;b++) tmp[++i]=ans[b];
         tmp[]=i;
         memcpy(ans,tmp,sizeof(ans));
     }else{
         int m=l+r>>;
         if(x<=m) query(p<<,l,m,x,y);
         if(y>=m+) query(p<<|,m+,r,x,y);
     }
 }

 void change(int p,int l,int r,int x,int y,bool b){
     if(l==r){
         if(!b) st[id[l]].erase(st[id[l]].find(y));
         else st[id[l]].insert(y);
         int i=;
         for(IT it=st[id[l]].begin();i<K&&it!=st[id[l]].end();it++){
             fr[p][++i]=*it;
         }fr[p][]=i;
     }else{
         int m=l+r>>;
         if(x<=m) change(p<<,l,m,x,y,b);
         else change(p<<|,m+,r,x,y,b);
         update(p);
     }
 }

 void solve1(int x,int y){
     ans[]=,tmp[]=;
     while(top[x]!=top[y]){
         if(dep[top[x]]<dep[top[y]]) swap(x,y);
         query(,,N,dfn[top[x]],dfn[x]);
         // printf("%d %d\n",dfn[top[x]],x);
         x=fa[top[x]];
     }
     if(dep[x]<dep[y]) swap(x,y);
     query(,,N,dfn[y],dfn[x]);
 }

 void solve2(int x,int y){
     change(,,N,dfn[smr[x][]],smr[x][],);
     change(,,N,dfn[y],smr[x][],);
     smr[x][]=y;
 }
 void solve3(int x,int y){
     change(,,N,dfn[smr[x][]],smr[x][],);
     change(,,N,dfn[smr[x][]],y,);
     smr[x][]=y;
 }

 int main(){
     //freopen("","r",stdin);
     int i,j,k;
     N=rd();
     for(i=;i<N;i++){
         int a=rd(),b=rd();
         adeg(a,b);adeg(b,a);
     }
     M=rd();
     for(i=;i<=M;i++){
         smr[i][]=rd(),smr[i][]=rd();
         st[smr[i][]].insert(smr[i][]);
     }
     Q=rd(),K=rd();
     dep[]=;dfs1();
     dfs2();build(,,N);
     for(j=;j<=Q;j++){
         int a=rd(),b=rd(),c=rd();
         if(a==){
             solve1(b,c);
             if(ans[]==) printf("-1");
             for(i=;i<=ans[];i++)
                 printf("%d ",ans[i]);
             printf("\n");
         }else if(a==) solve2(b,c);
         else solve3(b,c);
     }
     return ;
 }