hdu 4006 The kth great number(线段树 || 优先队列)

来源：互联网发布：paxos算法执行过程编辑：程序博客网时间：2024/05/20 18:51

题目：http://acm.hdu.edu.cn/showproblem.php?pid=4006

求一个数字序列的第K大的值。先输入两个数字n,k,接着是n行输入，I表示加入新的数字，Q是询问第k大的数字。

Sample Input

8 3I 1I 2I 3QI 5QI 4Q

Sample Output

练习线段树遇到这道题的，听说方法还挺多的。
线段树：

#include <iostream>#include <cstdio>#include <algorithm>using namespace std;const int maxn=1e6+5; //',INF=0x3f3f3f3f;int ans[maxn];struct node{    int l,r,count;    int mid(){  return (l+r)/2; }    int length(){  return (r-l+1); }}tree[maxn<<2];void build(int a,int b,int root){    tree[root].l=a;    tree[root].r=b;    tree[root].count=0;    if(a==b){        return ;    }    int m=tree[root].mid();    build(a,m,2*root);    build(m+1,b,2*root+1);}void insert(int root,int number){    if(tree[root].l==tree[root].r){         tree[root].count++;         return ;    }    if(number<=tree[root].mid()) insert(2*root,number);    else insert(2*root+1,number); // number-tree[2*root].count is wrong    tree[root].count=tree[2*root].count+tree[2*root+1].count;}int query(int root,int number){    if(tree[root].l==tree[root].r){         return tree[root].l;    }    int ans;    if(number<=tree[2*root].count)ans=query(2*root,number);    else ans=query(2*root+1,number-tree[2*root].count);  //we need "number-tree[2*root].count"    return ans;}int main(){    //freopen("cin.txt","r",stdin);    int n,k;    while(cin>>n>>k){        build(1,maxn,1);        char str[3];        int top=0;        for(int i=0;i<n;i++){            scanf("%s",str);            if(str[0]=='I'){                int a;                scanf("%d",&a);                insert(1,a);                top++;            }            else {                 printf("%d\n",query(1,top-k+1));            }        }    }    return 0;}

优先队列：
第一次写好的代码怀着忐忑的心情交上去，没错，果然超时了：

#include <iostream>#include <cstdio>#include <queue>using namespace std;const int maxn=1e6+5;int number[maxn];int main(){    //freopen("cin.txt","r",stdin);    int n,k;    while(cin>>n>>k){        priority_queue<int> que;        char s[3];        for(int i=0;i<n;i++){            scanf("%s",s);            if(s[0]=='I'){                int a;                scanf("%d",&a);                que.push(a);            }            else {                int j;                for(j=0;j<k;j++){                    number[j]=que.top();                    que.pop();                }                printf("%d\n",number[j-1]);                for(j=0;j<k;j++){                    que.push(number[j]);                }            }        }    }    return 0;}

想个办法把第二个循环去掉吧，抓住主要矛盾，忽略次要因素：

#include <iostream>#include <cstdio>#include <queue>#include <vector>#include <functional>using namespace std;const int maxn=1e6+5;int number[maxn];int main(){    //freopen("cin.txt","r",stdin);    int n,k;    while(cin>>n>>k){        priority_queue<int,vector<int>,greater<int> > que;  //从小到大        char s[3];        for(int i=0;i<n;i++){            scanf("%s",s);            if(s[0]=='I'){                int a;                scanf("%d",&a);                que.push(a);                if(que.size()>k)que.pop();            }            else {                printf("%d\n",que.top());            }        }    }    return 0;}

记住要用functional头文件，不然会Compilation Error。

0 0