蓝桥杯——算法训练—— 区间k大数查询

蓝桥杯——算法训练—— 区间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。
——————————————————————————————
思路分析：这道题实际上非常简单，但是它事儿特多，稍微麻烦点的就是它要求输出数组指定区间指定位置的值，这个本质上就是数组移植（千万不要对原数组进行排序），还有一点需要注意的就是题目要求降序排列，我们需要自定义一个cmp。下面来看代码：

#include <stdio.h>
#include <string>
#include <string.h>
#include <algorithm>
using namespace std;
int a[1005];
bool cmp(int a, int b){
	return a>b;
}
int main(){
	int n;
	while(scanf("%d", &n) != EOF){
		memset(a, 0, sizeof(a)); // 初始化是好习惯
		for(int i=1; i<=n; i++) scanf("%d", &a[i]);
		int m;
		scanf("%d", &m);
		while(m--){ // 输入指令
			int l, r, k;
			scanf("%d%d%d", &l, &r, &k);
			int l_ans = r-l+1;
			int ans[l_ans]; // 用来移植数组的指定区间
			memset(ans, 0, sizeof(ans));
			for(int i=0; i<l_ans; i++) ans[i] = a[i+l]; // 移植数组的指定区间
			sort(ans, ans+l_ans, cmp); // 注意这里的cmp，用来实现sort的降序
			printf("%d\n", ans[k-1]);
		}
	}
	return 0;
}