首页 > 代码库 > ZOJ 3612 Median (multiset)
ZOJ 3612 Median (multiset)
The median of m numbers is after sorting them in order, the middle one number of them if m is even or the average number of the middle 2 numbers if m is odd. You have an empty number list at first. Then you can add or remove some number from the list.
For each add or remove operation, output the median of the number in the list please.
Input
This problem has several test cases. The first line of the input is an integer T (0<T<=100) indicates the number of test cases. The first line of each test case is an integer n(0<n<=10000) indicates the number of operations. Each of the next n lines is either "add x" or "remove x"(-231<=x<231) indicates the operation.
Output
For each operation of one test case: If the operation is add output the median after adding x in a single line. If the operation is remove and the number x is not in the list, output "Wrong!" in a single line. If the operation is remove and the number x is in the list, output the median after deleting x in a single line, however the list is empty output "Empty!".
Sample Input
2 7 remove 1 add 1 add 2 add 1 remove 1 remove 2 remove 1 3 add -2 remove -2 add -1
Sample Output
Wrong! 1 1.5 1 1.5 1 Empty! -2 Empty! -1
Hint
if the result is an integer DO NOT output decimal point. And if the result is a double number , DO NOT output trailing 0s.
分析:比赛时也想到了用multiset,但是删除时会把所有相同的元素一起删掉,没想到用指针指向要删除的元素。这题还可以用树状数组或者线段树来做。
#include<stdio.h> #include<string.h> #include<set> #include<algorithm> using namespace std; multiset<int> ms; multiset<int>::iterator it; //指向要删除的元素的位置 multiset<int>::iterator it1; //记录中位数的位置 multiset<int>::iterator it2; //辅助迭代器 int main() { int t, n, i; char op[20]; int a; scanf("%d",&t); while(t--) { ms.clear(); scanf("%d",&n); for(i = 0; i < n; i++) { scanf("%s%d",op, &a); if(!strcmp(op, "add")) { ms.insert(a); if(ms.size() == 1) it1 = ms.begin(); else { if((ms.size()&1) && a >= *it1) it1++; else if((ms.size() % 2 == 0 && a < *it1)) it1--; } } else { it = ms.find(a); if(it == ms.end()) { printf("Wrong!\n"); continue; } else { if(*it1 == a) { it = it1; if(ms.size() % 2 == 0) it1++; else it1--; } else { if((ms.size()&1) && a >= *it1) it1--; else if(ms.size() % 2 == 0 && a < *it1) it1++; } ms.erase(it); } } int s = ms.size(); if(s == 0) printf("Empty!\n"); else { if(s&1) printf("%d\n",*it1); else { it2 = it1; it2++; long long ans = (long long)*it1 + (long long)*it2; if(ans % 2 == 0) printf("%lld\n",ans/2); else printf("%.1lf\n",ans/2.0); } } } } return 0; }