2.Acwing基础课第786题-简单-第k个数

题目描述

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。

输入格式

第一行包含整数 n 和 k。

第二行包含 n 个整数（所有整数均在1~范围内），表示整个数列。

输出格式

输出一个整数，表示数列的第 k 小数。

数据范围

1≤n≤100000

输入样例

5 3
2 4 1 5 3

输出样例

思路解析：

算法：快速排序（ Quick Sort ）

时间复杂度：O(nlog(n))

解题思路：

1.使用快速排序，把数组从小到大进行排序
2.然后直接输出对应下标的数（记得减一）

代码：

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<string>
using namespace std;

const int N = 100010;

int n;
int q[N], tmp[N];

void merge_sort(int q[], int l, int r)
{
	if (l >= r) return;

	int mid = l + r >> 1;

	merge_sort(q, l, mid), merge_sort(q, mid + 1, r);

	int k = 0, i = l, j = mid + 1;
	while (i <= mid && j <= r)
	{
		if (q[i] <= q[j]) tmp[k++] = q[i++];
		else tmp[k++] = q[j++];
	}

	while (i <= mid) tmp[k++] = q[i++];
	while (j <= r) tmp[k++] = q[j++];

	for (i = l, j = 0; i <= r; i++, j++) q[i] = tmp[j];
}

int main()
{
	scanf("%d", &n);
	for (int i = 0; i < n; i++) scanf("%d", &q[i]);

	merge_sort(q, 0, n - 1);
	for (int i = 0; i < n; i++) printf("%d,", q[i]);

	system("pause");
	return EXIT_SUCCESS;
}