【Bzoj4336】騎士的旅行

題意

給一棵樹，每個結點上可能有多個權值，可以每次改變某權值的值或者位置，多次詢問一條路徑上的前k大值。

---

解析

對於每個結點維護一個multiset，因爲k很小，所以每次可以把兩結點暴力合併，這樣就只要把兩個隊列合併就可以了。

---

#include <set>
#include <cstdio>
#include <algorithm>

#define Rep( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i<=(i##_END);i++)
#define For( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i++)
#define Lop( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i>=(i##_END);i--)
#define Dnt( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i--)

using std :: max;
using std :: min;

const int maxx = 40000 + 25;

typedef int Array[maxx];

Array top,rnk,size,son,ftr,dpt,head,f,p;

int n,m,x,y,z,tot,num,t,k,opt,tis;
int to[maxx<<1],nxt[maxx<<1];

std :: multiset<int> S[maxx];
std :: multiset<int> :: iterator it;

class Node{
    public:
        int cnt;
        int val[25];
}T[maxx<<2];

Node ans;

namespace Cute{

    void Ins(int x,int y){
        to[++num] = y;nxt[num] = head[x];head[x] = num;
    }

    void Dfs(int x){
        size[x] = 1;
        for(int i=head[x];i;i=nxt[i]){
            if(to[i] == ftr[x]) continue;
            dpt[to[i]] = dpt[x] + 1;ftr[to[i]] = x;
            Dfs(to[i]);size[x] += size[to[i]];
            if(size[to[i]] > size[son[x]]) son[x] = to[i];
        }
    }

    void __Dfs(int x,int brn){
        top[x] = brn;rnk[x] = ++tot;
        if(son[x]) __Dfs(son[x],brn);
        for(int i=head[x];i;i=nxt[i])
            if(to[i] != ftr[x] && to[i] != son[x])
                __Dfs(to[i],to[i]);
    }

    Node upt(Node a,Node b){
        int kk = 0,l = 1,r = 1;
        Node ans;ans.cnt = 0;
        while(l <= a.cnt && r <= b.cnt && kk < k){
            if(a.val[l] >= b.val[r]) ans.val[++kk] = a.val[l++];
            else ans.val[++kk] = b.val[r++];
        }
        while(l <= a.cnt && kk < k) ans.val[++kk] = a.val[l++];
        while(r <= b.cnt && kk < k) ans.val[++kk] = b.val[r++];
        ans.cnt = kk;
        return ans;
    }

    void modify(int i,int pos,int l,int r,int f,int v){

        if(l == r){

            if(f == 1) S[l].insert(v);
            if(f == 2) it = S[l].find(v),S[l].erase(it);

            tis = 0;

            if(!S[l].empty()){
                it = S[l].end();it --;
                while(tis < k){
                    T[i].val[++tis] = *it;
                    if(it == S[l].begin()) break;
                    it --;
                }
            }
            T[i].cnt = tis;
            return;
        }
        int mid = (l+r) >> 1;
        if(pos <= mid) modify(i<<1,pos,l,mid,f,v);
        if(pos >  mid) modify(i<<1|1,pos,mid+1,r,f,v);
        T[i] = upt(T[i<<1],T[i<<1|1]);
    }

    Node Query(int i,int x,int y,int l,int r){
        if(x <= l && r <= y) return T[i];
        int mid = (l+r) >> 1;Node ans;ans.cnt = 0;
        if(x <= mid) ans = upt(ans,Query(i<<1,x,y,l,mid));
        if(y >  mid) ans = upt(ans,Query(i<<1|1,x,y,mid+1,r));
        return ans;
    }

    Node Get(int x,int y){
        Node ans;ans.cnt = 0;
        while(top[x] != top[y]){
            if(dpt[top[x]] > dpt[top[y]]) std :: swap(x,y);
            ans = upt(ans,Query(1,rnk[top[y]],rnk[y],1,n));
            y = ftr[top[y]];
        }
        if(rnk[x] > rnk[y]) std :: swap(x,y);
        ans = upt(ans,Query(1,rnk[x],rnk[y],1,n));
        return ans;
    }

}

using namespace Cute;

int main(){
    scanf("%d",&n);
    For( i , 1 , n ) scanf("%d%d",&x,&y),Ins(x,y),Ins(y,x);
    Dfs(1);__Dfs(1,1);
    scanf("%d",&m);
    Rep( i , 1 , m ) scanf("%d%d",&f[i],&p[i]);
    scanf("%d%d",&t,&k);
    Rep( i , 1 , m ) modify(1,rnk[p[i]],1,n,1,f[i]);
    while( t-- ){
        scanf("%d%d%d",&opt,&x,&y);
        if(opt == 1){
            ans = Get(x,y);
            if(!ans.cnt) puts("-1");
            else{
                Rep( i , 1 , ans.cnt ) printf("%d ",ans.val[i]); 
                putchar(10);
            }
        }
        if(opt == 2){
            modify(1,rnk[p[x]],1,n,2,f[x]);
            p[x] = y;
            modify(1,rnk[p[x]],1,n,1,f[x]);
        }
        if(opt == 3){
            modify(1,rnk[p[x]],1,n,2,f[x]);
            f[x] = y;
            modify(1,rnk[p[x]],1,n,1,f[x]);
        }
    }
    return 0;
}

又變菜了。