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: [email protected] 
    > 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;
}