首页 > 代码库 > SPOJ 4487. Can you answer these queries VI splay
SPOJ 4487. Can you answer these queries VI splay
题目链接:点击打开链接
题意比较明显,不赘述。
删除时可以把i-1转到根,把i+1转到根下
则i点就在 根右子树 的左子树,且只有i这一个 点
#include<stdio.h> #include<iostream> #include<string.h> #include<algorithm> using namespace std; #define N 300500 #define inf 10000000 #define L(x) tree[x].ch[0] #define R(x) tree[x].ch[1] #define Father(x) tree[x].fa #define Size(x) tree[x].size #define Val(x) tree[x].val #define Left(x) tree[x].left #define Right(x) tree[x].right #define Sum(x) tree[x].sum #define Ans(x) tree[x].ans struct node{ int ch[2],fa,size; int val, left, right, sum, ans; }tree[N]; int tot, root; int a[N]; void Newnode(int& id, int val, int fa){ node E={0,0,fa,1,val,val,val,val,val}; id = tot++; tree[id] = E; } void push_up(int id){ Size(id) = Size(L(id))+Size(R(id))+1; Sum(id) = Sum(L(id)) + Sum(R(id)) + Val(id); Left(id) = max(Left(L(id)), Sum(L(id))+Val(id)+max(Left(R(id)), 0)); Right(id) = max(Right(R(id)), Sum(R(id))+Val(id)+max(Right(L(id)), 0)); Ans(id) = max(Ans(L(id)), Ans(R(id))); Ans(id) = max(Ans(id), Val(id)+max(Right(L(id)),0)+max(Left(R(id)),0)); } void push_down(int id){} void Rotate(int id, int kind){ int y = Father(id); push_down(id); push_down(y); tree[y].ch[kind^1] = tree[id].ch[kind]; Father(tree[id].ch[kind]) = y; if(Father(y)) tree[Father(y)].ch[R(Father(y))==y] = id; Father(id) = Father(y); Father(y) = id; tree[id].ch[kind] = y; push_up(y); } void Splay(int id, int goal){ push_down(id); while(Father(id)!=goal){ int y = Father(id); if(Father(y)==goal) Rotate(id,L(y)==id); else { int kind = L(Father(y))==y; if(tree[y].ch[kind]==id) { Rotate(id, kind^1); Rotate(id, kind); } else { Rotate(y, kind); Rotate(id, kind); } } push_down(id); } if(goal==0)root = id; push_up(id); } int Get_kth(int k){ int id = root; push_down(id); while(Size(L(id))!=k){ if(Size(L(id))>k) id = L(id); else { k -= (Size(L(id)) +1); id = R(id); } push_down(id); } return id; } int Getmax(int id){ push_down(id); while(R(id)){ id = R(id); push_down(id); } return id; } void Delete(int id){ int a = Get_kth(id-1); int b = Get_kth(id+1); Splay(a, 0); Splay(b, root); L(b) = 0; push_up(b); push_up(a); } int build(int l, int r, int& id, int fa){ if(l>r)return 0; int mid = (l+r)>>1; Newnode(id, a[mid], fa); build(l, mid-1, L(id), id); build(mid+1, r, R(id), id); push_up(id); } int n, que; char s[2]; void init(){ Father(0) = L(0) = R(0) = Size(0) = 0; Sum(0) = 0; Val(0) = Left(0) = Right(0) = Ans(0) = -inf; root = tot = 1; Newnode(root, -inf, 0); Newnode(R(root), -inf, root); build(1, n, L(R(root)), R(root)); push_up(R(root)); push_up(root); } int main(){ int i, j; while(~scanf("%d",&n)){ for(i = 1; i <= n; i++)scanf("%d",&a[i]); init(); scanf("%d",&que); while(que--){ scanf("%s",s); if(s[0]==‘I‘) { scanf("%d %d",&i,&j); Splay(Get_kth(i), 0); Splay(Getmax(L(root)), root); Newnode(R(L(root)), j, L(root)); push_up(L(root)); push_up(root); } else if(s[0]==‘Q‘) { scanf("%d %d",&i,&j); Splay(Get_kth(i-1), 0); Splay(Get_kth(j+1), root); printf("%d\n", Ans(L(R(root)))); } else if(s[0]==‘D‘) { scanf("%d",&i); Delete(i); } else if(s[0]==‘R‘) { scanf("%d %d",&i,&j); Splay(Get_kth(i), 0); Val(root) = j; push_up(root); } } } return 0; } /* 5 3 -4 3 -1 6 99 I 6 2 Q 3 5 R 5 -4 Q 3 5 D 2 Q 1 5 I 2 -10 Q 1 6 R 2 -1 Q 1 6 Q 1 6 */
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。