BZOJ1588 [HNOI2002] 营业额统计

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <algorithm>
 4 #include <cstring>
 5 #include <ctime>
 6 #define rep(i,l,r) for(int i=l; i<=r; i++)
 7 #define clr(x,y) memset(x,y,sizeof(x))
 8 typedef long long ll;
 9 typedef unsigned long long ull;
10 using namespace std;
11 const int INF = 0x3f3f3f3f;
12 const int maxn = 50010;
13 struct node{
14     int v,w,l,r,rnd,size;
15 }t[maxn];
16 int n,x,t1,t2,tot=0,root=0,ans=0;
17 inline int read(){
18     int ans = 0, f = 1;
19     char c = getchar();
20     while (!isdigit(c)){
21         if (c == ‘-‘) f = -1;
22         c = getchar();
23     }
24     while (isdigit(c)){
25         ans = ans * 10 + c - ‘0‘;
26         c = getchar();
27     }
28     return ans * f;
29 }
30 inline void update(int w){
31     t[w].size = t[t[w].l].size + t[t[w].r].size + t[w].w;
32 }
33 void rotl(int &w){
34     int k = t[w].r; t[w].r = t[k].l; t[k].l = w;
35     update(w); update(k); w = k;
36 }
37 void rotr(int &w){
38     int k = t[w].l; t[w].l = t[k].r; t[k].r = w;
39     update(w); update(k); w = k;
40 }
41 void insert(int x,int &w){
42     if (!w){
43         w = ++tot; t[w].v = x; t[w].size = t[w].w = 1;
44         t[w].rnd = rand(); t[w].l = t[w].r = 0; return;
45     }
46     t[w].size++; if (t[w].v == x) t[w].w++;
47     else if (x < t[w].v){
48         insert(x,t[w].l);
49         if (t[t[w].l].rnd < t[w].rnd) rotr(w);
50     }
51     else{
52         insert(x,t[w].r);
53         if (t[t[w].r].rnd < t[w].rnd) rotl(w);
54     }
55 }
56 void bef(int x,int w){
57     if (!w) return;
58     if (t[w].v <= x){
59         t1 = t[w].v;
60         bef(x,t[w].r);
61     }
62     else bef(x,t[w].l);
63 }
64 void aft(int x,int w){
65     if (!w) return;
66     if (t[w].v >= x){
67         t2 = t[w].v;
68         aft(x,t[w].l);
69     }
70     else aft(x,t[w].r);
71 }
72 int main(){
73     n = read();
74     rep(i,1,n){
75         if (scanf("%d",&x) == EOF) x = 0;
76         t1 = -INF; t2 = INF;
77         bef(x,root); aft(x,root);
78         if (i == 1) ans += x;
79         else ans += min(x-t1,t2-x);
80         insert(x,root);
81     }
82     printf("%d\n",ans);
83     return 0;
84 }