题意
给一棵树,每个结点上可能有多个权值,可以每次改变某权值的值或者位置,多次询问一条路径上的前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;
}
又变菜了。