算法训练 区间k大数查询
时间限制:1.0s 内存限制:256.0MB
问题描述
给定一个序列,每次询问序列中第l个数到第r个数中第K大的数是哪个。
输入格式
第一行包含一个数n,表示序列长度。
第二行包含n个正整数,表示给定的序列。
第三个包含一个正整数m,表示询问个数。
接下来m行,每行三个数l,r,K,表示询问序列从左往右第l个数到第r个数中,从大往小第K大的数是哪个。序列元素从1开始标号。
输出格式
总共输出m行,每行一个数,表示询问的答案。
样例输入
5
1 2 3 4 5
2
1 5 2
2 3 2
样例输出
4
2
数据规模与约定
对于30%的数据,n,m<=100;
对于100%的数据,n,m<=1000;
保证k<=(r-l+1),序列中的数<=106。
本来以为又要用线段树,但仔细一看要求得数还不是最大数,用线段树的的话就不好弄了,加上数据量很小,所以直接用数组存了,然后突发奇想,前阵子刚学了STL,set这个容器的特点:首先容器里的数值是有序的,然后,它可以使用数组进行初始化,这样就可以很方便的得到一段有序的数值,正好符合题目要求,如果数据降序排列,那么只要输出第k个元素就可以了,但有一点不方便的就是读取第k个数据需要使用迭代器而不能使用数组下标,还有就是需要自定义一个比较器来实现降序排列,还有就是必须用multiset才能实现容器中有重复元素,下面看代码
#include<iostream>
#include<set>
using namespace std;
struct cmp
{
bool operator()(const int&a,const int&b)
{
return a>b;
}
};
int main()
{
int n,i,cas,l,r,k;
cin>>n;
int *num=new int[n];
for(i=0;i<n;i++)
{
cin>>num[i];
}
cin>>cas;
while(cas--)
{
cin>>l>>r>>k;
multiset<int,cmp> s(num+l-1,num+r);
multiset<int,cmp>::iterator it;
/*for(it=s.begin();it!=s.end();it++)
{
cout<<*it<<' ';
}
cout<<endl;
*/
for(it=s.begin();k>0;k--,it++){}
it--;
cout<<*it<<endl;
}
return 0;
}