2021-12-28每日刷题打卡

一、Y总视频进度

二、刷题

2.1 AcWing-786：第K个数

1. 问题描述

2. 问题分析

基于快排的思想

• 首先还是先找一个x，确定好边界l， r，找到x所在的位置
• 然后看第k小的数在x的左边还是右边，进行递归

3. 问题解决

#include <iostream>
using namespace std;

const int N = 100010;

int q[N];
int n, k;

int q_sort(int q[], int l, int r, int k)
{
	if(l >= r) return q[l];
	
	int x = q[l+r >> 1], i = l-1, j = 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]);
	}
	
	if(j-l+1 >= k) return q_sort(q, l, j, k);
	else return q_sort(q, j+1, r, k - (j-l+1));
}

int main()
{
	scanf("%d%d", &n, &k);	
	
	for(int i = 0; i < n; i++) scanf("%d", &q[i]);
	
	printf("%d\n", q_sort(q, 0, n-1, k)); 
	
	return 0;
}