蓝桥杯基础题 区间k大数查询 c++

问题描述

给定一个序列，每次询问序列中第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

---

思路：

没有将要比较的序列全部排序，用了类似栈的模式，限制数组长度，存入第1到第k大值进入数组，剩下的与数组内值比较，大于数组内第k大即满足条件（突然想到在进入数组时，从最小的开始比较是不是会节省时间，懒得想了就这样吧），不满足就不用进入数组，所以该数组最后一个值就是第k大值，存入结果数组即可。

---

代码：

#include <iostream>
using namespace std;
int a[1000] = { 0 };//存储序列
int b[1000][3] = {0};//存储查找信息
int res[1000];//结果数组

int main()
{
	int n;//输入序列长度
	cin >> n;
	int i;
	for (i = 0; i <n; i++)
	{
		cin >> a[i];//输入序列
	}
	int m;
	cin >> m;//询问个数
	for (i = 0; i < m; i++)
	{
        //一次循环完成一次结果存入
        //输入m行内每行l、r、k信息
		cin >> b[i][0];//l
		cin >> b[i][1];//r
		cin >> b[i][2];//k
        //定义数组q存储从第l到r个数字，并按从大到小的顺序排列
        //类似栈的用法，将最大的存储在栈底，依此网上排，长度不超过k(第k大值)
		int q[1000] = {0};
		int l;//表示是序列中的第l个数
		int base = 0;//类似栈的指针，记录当前数组长度
		for (l = b[i][0]; l <= b[i][1]; l++)//第l到r个数
		{
			if(base < b[i][2])
			{
                //在未达到指定长度时，按大小顺序存入
				int x;//用x取p数组的值进行比较大小
				int flag=0;//判断当前序列中第l个数是否进入p数组
				for (x = 0; x < base; x++)
				{
					if (a[l-1] > q[x])
					{
						//类似入栈操作
						int p[1000] = { 0};
						int j;
						int p1 = 0;
						for (j = base; j >= x; j--)
						{
							p[p1] = q[j];
							p1++;
						}
						q[x] = a[l-1];
						x = x + 1;
						for (p1=p1-1; p1 >= 0; p1--)
						{
							q[x] = p[p1];
							x++;
						}
						flag = 1;//存入数组后记录并退出for循环
						break;
					}
				}
				if (flag == 0)
				{
                    //没有存入且长度未到达指定长度k，说明该值为当前最小，存在p数组顶部
					q[base] = a[l-1];
				}
				base++;//base++
			}
			else
			{
                //base数等于k说明现在只需要找到p[base-1]处存储的数即可。
                //因此还未存入的数若大于当前p数组内最小值，可以进入数组，否则不用。
				int x;
				for (x = 0; x < base; x++)
				{
					if (a[l-1] > q[x])
					{
						//类似入栈操作，重复
						int p[1000] = {0};
						int j;
						int p1 = 0;
						for (j = base-1; j >= x; j--)
						{
							p[p1] = q[j];
							p1++;
						}
						q[x] = a[l-1];
						x = x + 1;
						for (p1=p1-1; p1 >= 0; p1--)
						{
							q[x] = p[p1];
							x++;
						}
						break;
					}
				}
			}
		}
		res[i] = q[b[i][2]-1];//存入结果数组
	}
	for (i = 0; i < m; i++)
	{
		cout << res[i] << endl;//输出结果
	}
	return 0;
}

提交结果：