題目意思：

? ? 有n種操作，I代表插入數據，q代表輸出第k大的數。

區間的大數據的操作，基本是用樹操作。

這到題目可以用三種方法來做。

（1） 線段樹

考慮到題目可能有負數，給每個值加500000直接貼代碼。

s[k].n代表在這個區間內的數的個數。

#include<cstdio> #include<iostream> #include<cstring> #include<cmath> using namespace std; int n,k; const int N=1500005; struct node {int l, r;int n; }s[N*4+1100];void bulid(int l,int r,int k) {if(l==r){s[k].l=l;s[k].r=r;s[k].n=0;return ;}s[k].l=l;s[k].r=r;s[k].n=0;int mid=(l+r)>>1;bulid(l,mid,k<<1);bulid(mid+1,r,k<<1|1);s[k].n=s[2*k].n+s[2*k+1].n; }void Insert(int d,int k) {if(s[k].l==s[k].r&&s[k].l==d){s[k].n++;return;}int mid=(s[k].l+s[k].r)/2;if(d<=mid) Insert(d,2*k);else Insert(d,2*k+1);s[k].n=s[2*k].n+s[2*k+1].n; }int query(int k,int a) {if(s[k].l==s[k].r){return s[k].l;}int ans;if(a<=s[k<<1].n) ans=query(k<<1,a);else if(a>s[k<<1].n) ans=query(k<<1|1,a-s[k<<1].n);s[k].n=s[2*k].n+s[2*k+1].n;return ans; } int main() {while(scanf("%d%d",&n,&k)!=EOF){bulid(1,N,1);getchar();int total=0;for(int i=1;i<=n;i++){char c;scanf("%c",&c);getchar();if(c=='I'){int tt;scanf("%d",&tt);getchar();Insert(tt+500000,1);total++;}else if(c=='Q'){printf("%d\n",query(1,total-k+1)-500000);}}} }
第二種方法：

用優先隊列。

優先隊列內部維護一個堆。

優先隊列中只要維護k個數就好。

#include<cstdio> #include<iostream> #include<cstring> #include<cmath> #include<algorithm> #include<vector> #include<queue> using namespace std; struct Node {int a;friend bool operator < (const struct Node n1,const struct Node n2){return n1.a>n2.a;} }; int main() {int n,k;while(scanf("%d%d",&n,&k)!=EOF){getchar();priority_queue<Node>q;for(int i=1;i<=n;i++){char c;scanf("%c",&c);getchar();if(c=='I'){Node tt;scanf("%d",&tt.a);getchar();q.push(tt);while(q.size()>k){q.pop();}}else {printf("%d\n",q.top().a);}}} }
第三種方法是用set.

set內部維護一顆紅黑樹。

參見：http://blog.csdn.net/cnh294141800/article/details/21657969

#include<stdio.h> #include<iostream> #include<set> using namespace std; int main() {int n,i,j,m,k;while(scanf("%d %d",&n,&k)!=EOF){multiset<int>s;multiset<int>::iterator it;for(i=0;i<n;i++){char a[2];scanf("%s",a);if(a[0]=='I'){scanf("%d",&m);s.insert(m);if(s.size()>k)s.erase(s.begin());}else{printf("%d\n",*s.begin());}}}return 0; }

總結

以上是生活随笔為你收集整理的hdu 4006 The kth great number 线段树/优先队列/set的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。