hdu 5249 树状数组求第K大

KPI

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 362    Accepted Submission(s): 150

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
  
  

数的个数少，但是值比较大，所以离散化。用树状数组维护序列，在树状数组中求第K大。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define MAX 10005
#define mod 10000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)1
using namespace std;
typedef long long ll;
int bit[MAX];
int cnt;
int lowbit(int x) {
    return x&(-x);
}
int sum(int x) {
    int sum=0;
    while(x>0) {
        sum+=bit[x];
        x-=lowbit(x);
    }
    return sum;
}
void push(int pos,int num) {
    while(pos<=cnt) {
        bit[pos]+=num;
        pos+=lowbit(pos);
    }
}
int kth(int k) {  //第K大
    int num=0;
    int x=0;
    for(int i=1<<18; i>0; i>>=1) {
        num+=i;
        if(num>=cnt||x+bit[num]>=k)
            num-=i;
        else x+=bit[num];
    }
    return num+1;
}
char s[MAX][20];
queue<int> q;
int main() {
    int n,a[MAX],x,b[MAX];
    int cas=0;
    while(scanf("%d",&n)!=EOF) {
        while(!q.empty()) q.pop();
        cnt=1;
        CLR(bit,0);
        printf("Case #%d:\n",++cas);
        for(int i=1; i<=n; i++) {
            getchar();
            scanf("%s",s[i]);
            if(s[i][0]=='i') {
                scanf("%d",&x);
                b[cnt++]=x;
                a[i]=x;
            }
        }
        sort(b+1,b+cnt);
        for(int i=1; i<=n; i++) {
            if(s[i][0]=='i') {
                int x=lower_bound(b+1,b+cnt,a[i])-b;
                push(x,1);
                q.push(x);
            } else if(s[i][0]=='o') {
                int x=lower_bound(b+1,b+cnt,b[q.front()])-b;
                push(x,-1);
                q.pop();
            } else if(s[i][0]=='q') {
                printf("%d\n",b[kth(q.size()/2+1)]);
            }
        }
    }
    return 0;
}