P1923 【深基9.例4】求第 k 小的数

最新推荐文章于 2024-07-02 09:05:18 发布

听雨686

最新推荐文章于 2024-07-02 09:05:18 发布

阅读量360

点赞数 8

文章标签：算法

本文链接：https://blog.csdn.net/2302_80193662/article/details/136043531

版权

题目：

代码：

#include <bits/stdc++.h>

using namespace std;

int n, k, i, j;

const int N = 5e6 + 10;

int q[N];

void quick_sort(int l, int r)
{                          
	if (l >= r) return; 
	                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                  
	int i = l, j = r, mid = q[l + r >> 1];
	while (i <= j)
	{
		while (q[i] < mid) i++; 
	    while (q[j] > mid) j--;	
		if (i <= j) 
		{
			swap(q[i], q[j]);
			i++;
			j--;
		}
	}
	if (k <= j) quick_sort(l , j);
	else if (k >= j && k <= i) return;
	else if (k >= i) quick_sort(i , r);
}

int main()
{
	std::ios::sync_with_stdio(false);
	
	cin >> n >> k;
	
	for(int i = 0; i < n; i++) cin >> q[i];
	
    quick_sort(0, n - 1);
	cout << q[k];
	
	return 0;
}