ZOJ Monthly, June 2012 - B Median - zoj 3612

最近好久没做比赛，加上考试什么的，压力很大。这几天休息，正好抽个空做下浙大月赛。发现自己果然变水了。

不过说起来，zju的月赛难度也在上升。

这是比赛时候做的第一题，题目是说每次会有 移除x 和 添加x 的操作，问你每次操作后，这串数组的中位数是什么。

操作n <= 10000

我的想法就是一共出现的数字最多就n个，可以把数据离散掉，然后用线段树，把操作更新掉，可以储存数组现有的数字个数ans。每次查找就查找(ans/2+1)个数字是啥就可以了，如果ans是偶数，再找ans/2个数字，两个相加/2

这题目唯一我的疑问就是为啥要LL。。。

怎么看数据不会爆啊，可是交了几次都wa了，我就一狠心改了LL交了一把...

View Code

  1 #include<cstdio>

  2 #include<iostream>

  3 #include<algorithm>

  4 #include<cstring>

  5 #include<cmath>

  6 #include<map>

  7 #include<vector>

  8 using namespace std;

  9 #define N 10010

 10 #define M 10010

 11 #define inf 2000000000

 12 typedef long long LL;

 13 map<LL, LL> id;

 14 map<LL, LL> fid;

 15 struct node {

 16     LL a, time;

 17     char str[7];

 18 }w[N];

 19 bool cmp1(const node &a, const node &b) {

 20     return a.a < b.a;

 21 }

 22 bool cmp2(const node &a, const node &b) {

 23     return a.time < b.time;

 24 }

 25 struct node1 {

 26     LL l, r, num;

 27 }tr[N*10];

 28 void build(LL L, LL R, LL x) {

 29     tr[x].l = L;

 30     tr[x].r = R;

 31     tr[x].num = 0;

 32     if (L == R) return ;

 33     LL mid = (L + R) / 2;

 34     build(L, mid, x*2);

 35     build(mid+1, R, x*2+1);

 36 }

 37 LL flag, ans, temp;

 38 void update(LL ii, LL x, LL f) {

 39     if (tr[x].l == tr[x].r) {

 40         if (!f) {

 41             if (!tr[x].num) flag = 1;

 42             else {

 43                 ans--;

 44                 tr[x].num--;

 45             }

 46         }

 47         else {

 48             ans++;

 49             tr[x].num++;

 50         }

 51         return ;

 52     }

 53     LL mid = (tr[x].l + tr[x].r) / 2;

 54     if (mid >= ii) update(ii, x*2, f);

 55     else update(ii, x*2+1, f);

 56     tr[x].num = tr[x*2].num + tr[x*2+1].num;

 57 }

 58 void find(LL ii, LL x) {

 59     if (tr[x].l == tr[x].r) {

 60         temp += fid[tr[x].r];

 61         return ;

 62     }

 63     if (tr[x*2].num >= ii) find(ii, x*2);

 64     else find(ii-tr[x*2].num, x*2+1);

 65 }

 66 int main() {

 67     LL T, n;

 68     scanf("%lld", &T);

 69     while (T--) {

 70         fid.clear();

 71         id.clear();

 72         scanf("%lld", &n);

 73         for(LL i = 0; i < n; ++i) {

 74             w[i].time = i;

 75             scanf("%s%lld", w[i].str, &w[i].a);

 76         }

 77         sort(w, w + n, cmp1);

 78         LL cs = 1;

 79         for (LL i = 0; i < n; ++i) {

 80             if (id.find(w[i].a) == id.end()) {

 81                 fid[cs] = w[i].a;

 82                 id[w[i].a] = cs++;

 83             }

 84         }

 85         build(1, cs-1, 1);

 86         sort(w, w + n, cmp2);

 87         ans = 0;

 88         for (LL i = 0; i < n; ++i) {

 89             LL ii = id[w[i].a];

 90             temp = flag = 0;

 91             if (w[i].str[0] == 'r') update(ii, 1, 0);

 92             else update(ii, 1, 1);

 93             if (flag) printf("Wrong!\n");

 94             else if(!ans) printf("Empty!\n");

 95             else {

 96                 find(ans/2+1, 1);

 97                 if (ans%2) printf("%lld\n", temp);

 98                 else {

 99                     find(ans/2, 1);

100                     if (temp%2 == 0) printf("%lld\n", temp/2);

101                     else printf("%.1f\n", double(temp)/2);

102                 }

103             }

104         }

105     }

106     return 0;

107 }