算法训练 区间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。
分析:由于有多组询问,我们不能在每次查找时对原序列的a[l…r]进行排序并查询,这样会打乱原序列的元素顺序,造成后续查找的错误。
所以这里我用了一个辅助数组,将原序列L到R的元素拷贝进去,并对其进行排序输出。排序用的sort函数,很好用的一个东西,再次推荐!代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
int cmp(int a, int b)
{
return a < b;
}
int fun(int *num, int n, int l, int r, int k)
{
int i, j, len = r - l + 1;
int *temp = new int[len+1];
for(i = 1, j = l; i <= len; i++) //拷贝
{
temp[i] = num[j++];
}
sort(temp + 1, temp + len + 1, cmp);
i = temp[len-k+1];
delete temp;
return i;
}
int main()
{
int n, *num, m; //n:数列长度, num:数列头指针, m:m次查找
int i, j; //i:循环变量
int *l, *r, *k; //l, r, k:询问序列从左往右第l个数到第r个数中,从大往小第k大的数是哪个
//第一行输入
cin >> n;
//第二行输入
num = new int[n+1];
for(i = 1; i <= n; i++)
{
cin >> num[i];
}
//第三行输入
cin >> m;
//接下来的m行
l = new int[m];
r = new int[m];
k = new int[m];
for(i = 0; i < m; i++)
{
cin >> l[i] >> r[i] >> k[i];
}
//查找并输出
for(i = 0; i < m; i++)
{
cout << fun(num, n, l[i], r[i], k[i]) << endl;
}
delete num;
delete l;
delete r;
delete k;
return 0;
}