区间k大数查询
题目描述
给定一个序列,每次询问序列中第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
分析: 定义两个够长的数组,第一个放入初始,第二个放入每次不同的区域,使用sort排序,从大往小取值
#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
int a[1000001], b[1000001];
int main()
{
int n,m;
cin >> n;;
for (int i = 1; i <= n; i++) {
cin >> a[i];
}
cin >> m;
while (m--) {
int k = 0;
int x, y, z;
cin >> x >> y >> z;
for (int j = x; j <= y; j++) {
b[k++] = a[j];
}
sort(b, b + k);
cout << b[k - z] << endl;
}
return 0;
}
这样写空间换时间,内存太大。
改进:利用序列容器的优点代替大数组
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,m;
cin >> n;
int a[n + 1];
for (int i = 1; i <= n; i++) {
cin >> a[i];
}
cin >> m;
while (m--) {
int k = 0;
int x, y, z;
cin >> x >> y >> z;
vector<int>v;
for (int j = x; j <= y; j++) {
v.push_back(a[j]);
k++;
}
sort(v.begin(), v.end());
cout << v[k-z] << endl;
}
return 0;
}
学会了其他办法会继续改进,感谢观看