CF 295E Yaroslav and Points(Splay)
题目大意:
两个操作
1 id op 把id的位置+op
2 id op 查询在【id。op】之间的全部的数的差
思路:
关键是pushup函数。
自己退一下会发现。跟区间的总和,区间的节点个数有关。
比方假设左区间是 1 2 的话
右区间来一个 9
那么
就要加上
9-1+9-2
#include <cstdio> #include <iostream> #include <algorithm> #include <cstring> #include <set> #define inf 0x3f3f3f3f #define maxn 222222 #define keyTree (ch[ch[root][1]][0]) //当把l-1放在根节点 r+1放在根节点的右子树 //那么根节点的右子树的左子树就是[l,r] 这个区间的全部值 using namespace std; typedef long long LL; int S[maxn],que[maxn],ch[maxn][2],pre[maxn],siz[maxn]; int root,top1,top2; LL ans[maxn],val[maxn],a[maxn],b[maxn]; LL sum[maxn]; set <LL> tab; void Treaval(int x) { if(x) { Treaval(ch[x][0]); printf("%I64d ",val[x]); Treaval(ch[x][1]); } } void debug() { // printf("root=%d\n",root); Treaval(root); puts(""); } void New(int &x,int PRE,LL v) { if(top2)x=S[--top2]; else x=++top1; ch[x][0]=ch[x][1]=0; siz[x]=1; pre[x]=PRE; /*special*/ sum[x]=v; ans[x]=0; val[x]=v; } void pushup(int x)/*special*/ { siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1; sum[x]=sum[ch[x][0]]+sum[ch[x][1]]+val[x]; ans[x]=ans[ch[x][0]]+ans[ch[x][1]]+siz[ch[x][0]]*val[x]-sum[ch[x][0]]+sum[ch[x][1]]-val[x]*siz[ch[x][1]]+siz[ch[x][0]]*sum[ch[x][1]]-siz[ch[x][1]]*sum[ch[x][0]]; } void pushdown(int x)/*special*/ { } void build(int &x,int s,int e,int f) { if(s>e)return; int mid=(s+e)>>1; New(x,f,a[mid]); if(s<mid)build(ch[x][0],s,mid-1,x); if(e>mid)build(ch[x][1],mid+1,e,x); pushup(x); } void Rotate(int x,int kind) { int y=pre[x]; pushdown(x); pushdown(y); ch[y][!kind]=ch[x][kind]; pre[ch[x][kind]]=y; if(pre[y])ch[pre[y]][ch[pre[y]][1]==y]=x; pre[x]=pre[y]; ch[x][kind]=y; pre[y]=x; pushup(y); } void Splay(int x,int goal) { pushdown(x); while(pre[x]!=goal) { if(pre[pre[x]]==goal) Rotate(x,ch[pre[x]][0]==x); else { int y=pre[x]; int kind=ch[pre[y]][0]==y; if(ch[y][kind]==x){ Rotate(x,!kind); Rotate(x,kind); } else { Rotate(y,kind); Rotate(x,kind); } } } pushup(x); if(goal==0)root=x; } void RotateTo(int k,int goal) { int r=root; pushdown(r); while(siz[ch[r][0]]!=k) { if(k<siz[ch[r][0]]) { r=ch[r][0]; } else { k-=siz[ch[r][0]]+1; r=ch[r][1]; } pushdown(r); } Splay(r,goal); } void erase(int x) { int y=pre[x]; int head=0,tail=0; for(que[tail++]=x;head<tail;head++) { S[top2++]=que[head]; if(ch[que[head]][0])que[tail++]=ch[que[head]][0]; if(ch[que[head]][1])que[tail++]=ch[que[head]][1]; } ch[y][ch[y][1]==x]=0; pushup(y); } void init(int n) { root=top1=top2=0; ch[0][0]=ch[0][1]=siz[0]=pre[0]=0; ans[0]=sum[0]=val[0]=0; New(root,0,-inf); New(ch[root][1],root,inf); siz[root]=2; for(int i=0;i<n;i++) { scanf("%I64d",&a[i]); tab.insert(a[i]); b[i]=a[i]; } tab.insert(-inf);tab.insert(inf); sort(a,a+n); build(keyTree,0,n-1,ch[root][1]); pushup(ch[root][1]); pushup(root); } int find(LL x,int t) { if(x==val[t])return t; else if(x>val[t])return find(x,ch[t][1]); else return find(x,ch[t][0]); } set<LL>::iterator it; int add(int x,LL Num,int pos) { if(Num<=val[x]) { if(ch[x][0]==0) { Splay(x,0); int S=siz[ch[root][0]]; RotateTo(S-1,0); RotateTo(S,root); New(keyTree,ch[root][1],Num); pushup(ch[root][1]); pushup(root); } else add(ch[x][0],Num,pos); } else { if(ch[x][1]==0) { Splay(x,0); int S=siz[ch[root][0]]; RotateTo(S,0); RotateTo(S+1,root); New(keyTree,ch[root][1],Num); pushup(ch[root][1]); pushup(root); } else add(ch[x][1],Num,siz[ch[x][0]]+1+pos); } } int main() { tab.clear(); int n; scanf("%d",&n); init(n); //debug(); int m; scanf("%d",&m); while(m--) { int op; int l,r; scanf("%d%d%d",&op,&l,&r); if(op==1) { it=tab.lower_bound(b[l-1]); //printf("*it = %I64d\n",*it); int pos=find(*it,root); //printf("---%I64d\n",val[pos]); Splay(pos,0); int Spos=siz[ch[root][0]]; RotateTo(Spos-1,0); RotateTo(Spos+1,root); //erase(keyTree); erase(keyTree); pushup(ch[root][1]); pushup(root); tab.erase(it); b[l-1]+=r; tab.insert(b[l-1]); add(root,b[l-1],0); } else { it=tab.lower_bound(l); it--; //printf("%I64d\n",*it); //printf("%I64d\n",val[]); Splay(find(*it,root),0); it=tab.upper_bound(r); Splay(find(*it,root),root); printf("%I64d\n",ans[keyTree]); } //debug(); } return 0; }