hdu5249---KPI(二分+树状数组)

Problem Description
你工作以后, KPI 就是你的全部了. 我开发了一个服务，取得了很大的知名度。数十亿的请求被推到一个大管道后同时服务从管头拉取请求。让我们来定义每个请求都有一个重要值。我的KPI是由当前管道内请求的重要值的中间值来计算。现在给你服务记录，有时我想知道当前管道内请求的重要值得中间值。

Input
有大约100组数据。

每组数据第一行有一个n(1≤n≤10000)，代表服务记录数。

接下来有n行，每一行有3种形式
“in x”: 代表重要值为x(0≤x≤109)的请求被推进管道。
“out”: 代表服务拉取了管道头部的请求。
“query: 代表我想知道当前管道内请求重要值的中间值. 那就是说，如果当前管道内有m条请求, 我想知道，升序排序后第floor(m/2)+1th 条请求的重要值.

为了让题目简单，所有的x都不同，并且如果管道内没有值，就不会有”out”和”query”操作。

Output
对于每组数据，先输出一行

Case #i:
然后每一次”query”，输出当前管道内重要值的中间值。

Sample Input

6
in 874
query
out
in 24622
in 12194
query

Sample Output

Case #1:
874
24622

Source
2015年百度之星程序设计大赛 - 初赛(1)

Recommend
hujie | We have carefully selected several similar problems for you: 5251 5250 5247 5244 5243

用个树状数组来求中位数就行了,二分一下
然后in out 的时候更新下树状数组

/*************************************************************************
    > File Name: 1004.cpp
    > Author: ALex
    > Mail: zchao1995@gmail.com 
    > Created Time: 2015年05月30日 星期六 17时32分56秒
 ************************************************************************/

#include <functional>
#include <algorithm>
#include <iostream>
#include <fstream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <queue>
#include <stack>
#include <map>
#include <bitset>
#include <set>
#include <vector>

using namespace std;

const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double eps = 1e-15;
typedef long long LL;
typedef pair <int, int> PLL;

static const int N = 10010;
int tree[N];
int xis[N];
int cnt;
int Stack[N << 1];
struct QUE {
    int flag;
    int op_num;
}que[N];

int BinSearch(int val) {
    int l = 1, r = cnt;
    int mid;
    while (l <= r) {
        mid = (l + r) >> 1;
        if (xis[mid] == val) {
            break;
        }
        else if (xis[mid] > val) {
            r = mid - 1;
        }
        else {
            l = mid + 1;
        }
    }
    return mid;
}

int lowbit(int x) {
    return x & (-x);
}

void add(int x, int val, int n) {
    for (int i = x; i <= n; i += lowbit(i)) {
        tree[i] += val;
    }
}

int sum(int x) {
    int ans = 0;
    for (int i = x; i; i -= lowbit(i)) {
        ans += tree[i];
    }
    return ans;
}

char str[20];

int main() {
    int n;
    int icase = 1;
    while (~scanf("%d", &n)) {
        cnt = 0;
        int cur;
        memset(tree, 0, sizeof(tree));
        for (int i = 1; i <= n; ++i) {
            scanf("%s", str);
            if (str[0] == ‘i‘) {
                scanf("%d", &cur);
                xis[++cnt] = cur;
                que[i].op_num = cur;
                que[i].flag = 0; // in num
            }
            else if (str[0] == ‘o‘) {
                que[i].flag = 1; // out
            }
            else {
                que[i].flag = 2; // query
            }
        }
        sort(xis + 1, xis + cnt + 1);
        cnt = unique(xis + 1, xis + 1 + cnt) - xis - 1;
        printf("Case #%d:\n", icase++);
        int s = 0, e = -1, res = 0;
        for (int i = 1; i <= n; ++i) {
            if (que[i].flag == 0) {
                ++res;
                int val = BinSearch(que[i].op_num);
                Stack[++e] = val;
                add(val, 1, cnt);
            }
            else if (que[i].flag == 1) {
                --res;
                int val = Stack[s];
                add(val, -1, cnt);
                ++s;
            }
            else {
                int l = 1, r = cnt, mid, ans;
                while (l <= r) {
                    mid = (l + r) >> 1;
                    int les = sum(mid - 1);
                    int equ = sum(mid);
                    int m = (res >> 1) + 1;
                    if (les < m && equ >= m) {
                        ans = mid;
                        break;
                    }
                    else if (equ < m) {
                        l = mid + 1;
                    }
                    else {
                        r = mid - 1;
                    }
                }
                printf("%d\n", xis[ans]);
            }
        }
    }
    return 0;
}