[C++] AcWing 786. 第k个数

前言

只是记录

思路

• 先将数据从小到大排序——快速排序（直接用模板）
• 查找第k个数
• 查找的时候注意数组下标（从0开始，k-1为结果；从1开始，k为结果）

代码

#include<bits/stdc++.h>
using namespace std;
const int N=100010;
void quick_sort(int q[], int l, int r)//模板
{
	if(l>=r)
		return;
	int i=l-1,j=r+1,x=q[l+r>>1];
	while(i<j)
	{
		do i++; while(q[i]<x);
		do j--; while(q[j]>x);
		if(i<j) swap(q[i],q[j]);
	}
	quick_sort(q,l,j);
	quick_sort(q,j+1,r);
}
int main()
{
	int n,k;
	cin>>n>>k;
	int q[N];
	for(int i=0;i<n;i++)
		cin>>q[i];
	quick_sort(q,0,n-1);
	cout<<q[k-1];//注意下标
	return 0;
}

题目链接

786. 第k个数