[蓝桥杯]-区间k大数查询
环境:Dev-C++ 5.9.2
编译器:TDM-GCC 4.8.1
问题描述
给定一个序列,每次询问序列中第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。
#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
using namespace std;
bool cmp(int a,int b)
{
return a>b;
}//自定义比较函数,返回较大的数配合sort函数使用
int main()
{
int n,i,j,m,l,r,k;
cin>>n;
int arr[n],t[n];
for(i=0;i<n;i++)
cin>>arr[i];
cin>>m;
while(m)
{
cin>>l>>r>>k;
for(j=0,i=l-1;i<r,j<r-l+1;i++,j++)
t[j]=arr[i];//选出的数存放在t数组中,方便后面的排序,主要是为了不改变原数组序列
sort(t,t+j,cmp);//加了cmp函数,sort函数自动将t数组降序排列
cout<<t[k-1]<<endl;
for(i=0;i<j;i++)
t[i]=0;//t数组的重置
m--;
}
return 0;
}
[总结]:主要用到了C++中的STL(Standard Template Library,标准模板库)中的重要函数之一排序函数sort(arr,arr+len)。当然这是原始的升序排列的函数,本题要求取第k大的数,所以加了一个自定义布尔型函数 bool cmp(int a,int b){ return a>b; },来实现数组的降序排列,方便后面的取数。当然,依题意也可以不加自定义函数,直接使用原sort函数,从数组的后面往前取数,即arr[len-k]也是题解。