Tranvick

treap sum

Mar 6th, 2012
187
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.48 KB | None | 0 0
  1. #pragma comment(linker, "/STACK:65777216")
  2. #include <iostream>
  3. #include <iomanip>
  4. #include <cstdlib>
  5. #include <ctime>
  6. #include <cstdio>
  7. #include <algorithm>
  8. #include <cmath>
  9. #include <vector>
  10. #include <set>
  11. #include <stack>
  12. #include <map>
  13. #include <queue>
  14. #include <string>
  15. #include <memory.h>
  16. #include <iterator>
  17. #define y1 trololoy1
  18. #define y0 trololoy0
  19. #define mem(A,X) memset(A,X,sizeof(A))
  20. #define memo(A) memset(A,0,sizeof(A))
  21. #define forn(I,B) for (int I=1;I<=(B);I++)
  22. #define forg(H,V) for (int H=first[V];h;h=next[H])
  23. #define rep(I,B) for (int I=0;I<(B);I++)
  24. #define labs(X) (((X)>0)?(X):(-(X)))
  25. #define ropen(X) freopen(X,"r",stdin)
  26. #define wopen(X) freopen(X,"w",stdout)
  27. #define rwopen(X) freopen(X".in","r",stdin);freopen(X".out","w",stdout)
  28. #define pb push_back
  29. #define mp make_pair
  30. #define all(X) (X).begin(),(X).end()
  31. #define sqr(X) ((X)*(X))
  32.  
  33. using namespace std;
  34.  
  35. typedef pair <int,int> pii;
  36. typedef double ld;
  37. typedef long long ll;
  38. typedef pair <ll,ll> pll;
  39. typedef vector<int> vi;
  40. const int N=2222;
  41. const int INF=111111111;
  42. const double eps=1e-9;
  43. const double pi=3.14159265358979;
  44.  
  45. struct treap{
  46.     int ls,rs,x,y,cost,sum;
  47. } t[N];
  48.  
  49. int c,n,root,a[N],k;
  50.  
  51. inline int sumof(int v){
  52.     return v?t[v].sum:0;
  53. }
  54.  
  55. inline void recalc(int v){
  56.     if (!v) return;
  57.     t[v].sum=sumof(t[v].ls)+sumof(t[v].rs)+t[v].cost;
  58. }
  59.  
  60. void merge(int & x,int l,int r){
  61.     if (!l || !r){
  62.         x=l?l:r;
  63.         return;
  64.     }
  65.     if (t[l].y>t[r].y){
  66.         x=l;
  67.         merge(t[x].rs,t[x].rs,r);
  68.     } else {
  69.         x=r;
  70.         merge(t[x].ls,l,t[x].ls);
  71.     }
  72.     recalc(x);
  73. }
  74.  
  75. void split(int x,int key,int &l,int &r){
  76.     if (!x){
  77.         l=r=0;
  78.         return;
  79.     }
  80.     if (t[x].x<=key){
  81.         l=x;
  82.         split(t[x].rs,key,t[x].rs,r);
  83.     } else {
  84.         r=x;
  85.         split(t[x].ls,key,l,t[x].ls);
  86.     }
  87.     recalc(x);
  88. }
  89.  
  90. void insert(int &x){
  91.     if (!x){
  92.         x=c;
  93.         return;
  94.     }
  95.     if (t[c].y>t[x].y){
  96.         split(x,t[c].x,t[c].ls,t[c].rs);
  97.         x=c;
  98.     } else insert(t[c].x<t[x].x?t[x].ls:t[x].rs);
  99.     recalc(x);
  100. }
  101.  
  102. inline void ins(int key,int cost){
  103.     t[++c]=(treap){0,0,key,a[c],cost,cost};
  104.     insert(root);
  105. }
  106.  
  107. void erase(int &x,int key){
  108.     if (!x) return;
  109.     if (t[x].x==key) merge(x,t[x].ls,t[x].rs);
  110.     else erase(t[x].x<key?t[x].rs:t[x].ls,key);
  111.     recalc(x);
  112. }
  113.  
  114. void print(int x){
  115.     if (!x) return;
  116.     print(t[x].ls);
  117.     cout<<t[x].x<<" ";
  118.     print(t[x].rs);
  119. }
  120.  
  121. inline int query(int l,int r){
  122.     int L,M1,M2,R;
  123.     split(root,l-1,L,M1);
  124.     split(M1,r,M2,R);
  125.     int q=sumof(M2);
  126.     merge(M1,M2,R);
  127.     merge(root,L,M1);
  128.     return q;
  129. }
  130.  
  131. int main(){
  132.     cin>>n;
  133.     forn(i,n) a[i]=i;
  134.     random_shuffle(a+1,a+n+1);
  135.     forn(i,n){
  136.         int x,y;
  137.         cin>>x>>y;
  138.         ins(x,y);
  139.     }
  140.     int a,b;
  141.     cin>>a>>b;
  142.     cout<<query(a,b);
  143. }
Advertisement
Add Comment
Please, Sign In to add comment