这题的复杂度可以到达惊人的\(\log^4\)据说还能跑过去(差点没吓死我
直接二分+树剖树套树(\(n \log^4 n\))
一个\(\log\)也不少的4\(\log\)
但是我有个\(\log^3\)的树剖上面整体二分+线段树的做法
不过据说有个\(\log\)做法,我不会,反正我菜就是了
然后的话 就直接考虑个消除贡献…然而并不需要排序
权值线段树 就可以了
每次添加修改就
if(! op) {
int y = read() , z = read() ;
Q[++ tot] = { 0 , -1 , val[y] , y , 1 } ; val[y] = z ;
Q[++ tot] = { 0 , 1 , val[y] , y , 1 } ; b[++ len] = z ;
} else {
int y = read() , z = read() ; Q[++ tot] = { y , z , op , ++ pos , 2} ;
}
然后就成功完成了打消贡献以及添加贡献
然后整体二分板子即可
#include
#define int long long
#define rep(a , b , c) for(int a = b ; a <= c ; ++ a)
#define Rep(a , b , c) for(int a = b ; a >= c ; -- a)
#define go(u) for(int i = G.head[u] , v = G.to[i] , w = G.dis[i] ; i ; v = G.to[i = G.nxt[i]] , w = G.dis[i])
using namespace std ;
using ll = long long ;
using pii = pair < int , int > ;
using vi = vector < int > ;
int read() {
int x = 0 ; bool f = 1 ; char c = getchar() ;
while(c < 48 || c > 57) {
if(c == '-') f = 0 ;
c = getchar() ;
}
while(c > 47 && c < 58) {
x = (x << 1) + (x << 3) + (c & 15) ;
c = getchar() ;
}
return f ? x : -x ;
}
template void print(T x , char c = '\n') {
static char st[100] ; int stp = 0 ;
if(! x) putchar('0') ; if(x < 0) x = -x , putchar('-') ;
while(x) st[++ stp] = x % 10 ^ 48 , x /= 10 ;
while(stp) putchar(st[stp --]) ; putchar(c) ;
}
template void cmax(T & x , T y) { x < y ? x = y : 0 ; }
template void cmin(T & x , T y) { x > y ? x = y : 0 ; }
const int _N = 1e6 + 10 ;
struct Group {
int head[_N] , nxt[_N << 1] , to[_N] , dis[_N] , cnt = 1 ;
Group () { memset(head , 0 , sizeof(head)) ; }
void add(int u , int v , int w = 1) { nxt[++ cnt] = head[u] ; to[cnt] = v ; dis[cnt] = w ; head[u] = cnt ; }
} ;
const int N = 5e5 + 10 ;
typedef int arr[N] ;
Group G ;
int n , q , len = 0 , val[N] , b[N << 1] ;
arr fa , d , son , sz ;
void dfs(int u) {
sz[u] = 1 ; go(u) if(v ^ fa[u]) {
fa[v] = u , d[v] = d[u] + 1 , dfs(v) , sz[u] += sz[v] ;
if(sz[v] > sz[son[u]]) son[u] = v ;
}
}
arr top , id ;
int idx = 0 ;
void dfs(int u , int t) {
top[u] = t ;
id[u] = ++ idx ;
if(son[u]) dfs(son[u] , t) ;
go(u) if(v ^ fa[u] && v ^ son[u]) dfs(v , v) ;
}
struct Seg {
int sum[N << 2] ;
void change(int l , int r , int rt , int pos , int val) {
if(l == r) { sum[rt] += val ; return ; }
int mid = l + r >> 1 ;
if(pos <= mid) change(l , mid , rt << 1 , pos , val) ;
else change(mid + 1 , r , rt << 1 | 1 , pos , val) ;
sum[rt] = sum[rt << 1] + sum[rt << 1 | 1] ;
}
int query(int a , int b , int l , int r , int rt) {
if(a <= l && r <= b) return sum[rt] ;
int mid = l + r >> 1 , ans = 0 ;
if(a <= mid) ans += query(a , b , l , mid , rt << 1) ;
if(b > mid) ans += query(a , b , mid + 1 , r , rt << 1 | 1) ;
return ans ;
}
} t ;
int query_range(int x , int y) {
int ans = 0 ;
while(top[x] != top[y]) { if(d[top[x]] < d[top[y]]) swap(x , y) ; ans += t.query(id[top[x]] , id[x] , 1 , n , 1) ; x = fa[top[x]] ; }
if(d[x] > d[y]) swap(x , y) ; ans += t.query(id[x] , id[y] , 1 , n , 1) ;
return ans ;
}
struct Query {
int x , y , k , id , type ;
} Q[N] , q1[N] , q2[N] ;
int tot = 0 , num = 0 , pos = 0 ;
arr ans ;
void solve(int L , int R , int l , int r) {
if(L > R) return ;
if(l == r) {
rep(i , L , R) if(Q[i].type == 2) ans[Q[i].id] = l ; return ;
}
int mid = l + r >> 1 , cnt1 = 0 , cnt2 = 0 ;
rep(i , L , R) {
if(Q[i].type == 1) {
if(Q[i].k > mid) {
t.change(1 , n , 1 , id[Q[i].id] , Q[i].y) ; q2[++ cnt2] = Q[i] ;
} else q1[++ cnt1] = Q[i] ;
} else {
int res = query_range(Q[i].x , Q[i].y) ;
if(res >= Q[i].k) q2[++ cnt2] = Q[i] ;
else { Q[i].k -= res ; q1[++ cnt1] = Q[i] ; }
}
}
rep(i , 1 , cnt2) if(q2[i].type == 1) t.change(1 , n , 1 , id[q2[i].id] , -q2[i].y) ;
rep(i , 1 , cnt1) Q[L + i - 1] = q1[i] ; rep(i , 1 , cnt2) Q[L + cnt1 + i - 1] = q2[i] ;
solve(L , L + cnt1 - 1 , l , mid) ; solve(L + cnt1 , R , mid + 1 , r) ;
}
signed main() {
n = read() ; q = read() ;
rep(i , 1 , n) { val[i] = read() ; Q[++ tot] = { 0 , 1 , val[i] , i , 1 } ; b[++ len] = val[i] ; }
rep(i , 2 , n) { int x = read() , y = read() ; G.add(x , y) ; G.add(y , x) ; }
dfs(1) ; dfs(1 , 1) ;
rep(i , 1 , q) {
int op = read() ;
if(! op) {
int y = read() , z = read() ;
Q[++ tot] = { 0 , -1 , val[y] , y , 1 } ; val[y] = z ;
Q[++ tot] = { 0 , 1 , val[y] , y , 1 } ; b[++ len] = z ;
} else {
int y = read() , z = read() ; Q[++ tot] = { y , z , op , ++ pos , 2} ;
}
}
sort(b + 1 , b + len + 1) , len = unique(b + 1 , b + len + 1) - b - 1 ;
rep(i , 1 , tot) if(Q[i].type != 2) Q[i].k = lower_bound(b + 1 , b + len + 1 , Q[i].k) - b ;
solve(1 , tot , 0 , len + 1) ;
rep(i , 1 , pos) {
if(! ans[i]) puts("invalid request!") ;
else print(b[ans[i]]) ;
}
return 0 ;
}