Link:http://hihocoder.com/problemset/problem/1133
#1133 : 二分·二分查找之k小数
时间限制:
10000ms
单点时限:
1000ms
内存限制:
256MB
-
10 4 1732 4176 2602 6176 1303 6207 3125 1 1011 6600
样例输出
-
1732
描述
在上一回里我们知道Nettle在玩《艦これ》,Nettle的镇守府有很多船位,但船位再多也是有限的。Nettle通过捞船又出了一艘稀有的船,但是已有的N(1≤N≤1,000,000)个船位都已经有船了。所以Nettle不得不把其中一艘船拆掉来让位给新的船。Nettle思考了很久,决定随机选择一个k,然后拆掉稀有度第k小的船。 已知每一艘船都有自己的稀有度,Nettle现在把所有船的稀有度值告诉你,希望你能帮他找出目标船。
输入
第1行:2个整数N,k。N表示数组长度,
第2行:N个整数,表示a[1..N],保证不会出现重复的数,1≤a[i]≤2,000,000,000。
输出
第1行:一个整数t,表示t在数组中是第k小的数,若K不在数组中,输出-1。
方法一:优先队列
AC code1:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<queue>
#include<map>
#include<cmath>
using namespace std;
struct node{
int v;
friend bool operator < (node a,node b)
{
return a.v<b.v;//大的优先
}
};
int main()
{
int n,k;
//freopen("D:\\in.txt","r",stdin);
while(scanf("%d%d",&n,&k)!=EOF)
{
int cnt=0;
priority_queue<node>pq;
while(n--)
{
//char c;
//cin>>c;
//if(c=='I')
//{
if(cnt<k)
{
node t;
scanf("%d",&t.v);
pq.push(t);
cnt++;
}
else
{
node t;
scanf("%d",&t.v);
if(pq.top().v>t.v)
{
pq.pop();
pq.push(t);
}
}
//}
/*else
{
if(c=='Q')
{*/
/* }
}*/
}
printf("%d\n",pq.top().v);
}
return 0;
}
方法二:SBT
AC code2:
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#include<vector>
using namespace std;
const int N=1000005;
struct TNode {
int left,right,size,key;
void init() {
left=0; right=0; size=1;
}
}node[N];
int n,m,tot,root;
void leftrorate(int &t) {//左旋
int k=node[t].right;
node[t].right=node[k].left;
node[k].left=t;
node[k].size=node[t].size;
node[t].size=node[node[t].left].size+node[node[t].right].size+1;
t=k;
}
void rightrorate(int &t) {//右旋
int k=node[t].left;
node[t].left=node[k].right;
node[k].right=t;
node[k].size=node[t].size;
node[t].size=node[node[t].left].size+node[node[t].right].size+1;
t=k;
}
void maintain(int &t,bool flag) {//维护
if(!flag) {
if(node[node[node[t].left].left].size>node[node[t].right].size)
rightrorate(t);
else if(node[node[node[t].left].right].size>node[node[t].right].size) {
leftrorate(node[t].left);
rightrorate(t);
}
else return ;
}
else {
if(node[node[node[t].right].right].size>node[node[t].left].size)
leftrorate(t);
else if(node[node[node[t].right].left].size>node[node[t].left].size) {
rightrorate(node[t].right);
leftrorate(t);
}
else return ;
}
maintain(node[t].left,0);
maintain(node[t].right,1);
maintain(t,0);
maintain(t,1);
}
void insert(int &t,int value) {//插入
if(!t) {
t=++tot;
node[t].init();
node[t].key=value;
}
else {
node[t].size++;
if(value<node[t].key) insert(node[t].left,value);
else insert(node[t].right,value);
maintain(t,value>=node[t].key);
}
}
int find(int t,int k) {//查找第k小的数
if(k<=node[node[t].left].size) return find(node[t].left,k);
else if(k>node[node[t].left].size+1)
return find(node[t].right,k-node[node[t].left].size-1);
else return node[t].key;
}
int main() {
//freopen("D:\\in.txt","r",stdin);
int k,end;
char le[5];
while(~scanf("%d%d",&n,&end)) {
root=tot=0;
for(k=0;k<n;k++) {
//scanf("%s",le);
le[0]='I';
switch(le[0]) {
case 'I':scanf("%d",&m);
insert(root,m); break;
default :
printf("%d\n",find(root,tot+1-end));
}
}
printf("%d\n",find(root,end));
}
return 0;
}