C++快速排序

C++快速排序

思路

首先随机选择一个元素作为pivot（k），然后将数组a分成三个部分：小于pivot的部分放在数组b中，等于pivot的部分放在数组c中，大于pivot的部分放在数组d中。然后，递归地对数组b和数组d进行快速排序，最后将三个数组合并起来得到排序后的数组a。

代码

#include <bits/stdc++.h>
using namespace std;
int a[20010];
int b[20010];
int c[20010];
int d[20010];
int get_rand(int l, int r)
{
	return (rand() % (r-l+1)) + l;
}
void quick(int l, int r)
{
	if(l >= r) return;
	int k = get_rand(l, r) ;
	int tot1 = 0, tot2 = 0, tot3 = 0;
	for(int i = l; i <= r; i++)
	{
		if(a[i] < a[k])
		{
			b[tot1++] = a[i];
		}
		else if(a[i] == a[k])
		{
			c[tot2++] = a[i];
		}
		else
		{
			d[tot3++] = a[i];
		}
		//根据K把 大于 等于 小于 分到 bcd 三个数组里面 
	}
	//将分好的数放回去
	for(int i = 0; i < tot1; i++) 
	{
		a[l + i] = b[i];
	}
	for(int i = 0; i < tot2; i++) 
	{
		a[l + tot1 + i] = c[i];
	}
	for(int i = 0; i < tot3; i++) 
	{
		a[l + tot1 + tot2 + i] = d[i];
	}
	quick(l, l + tot1 - 1);
	quick(l + tot1 + tot2, r);
	//递归 
}
int main()
{
	srand(time(0));
	int n;
	cin >> n;
	for(int i=1;i<=n;i++) cin >> a[i];
	quick(1, n);
	for(int i=1;i<=n;i++) cout << a[i] << ' ';
	return 0;
}

生成随机数

int get_rand(int l, int r)
{
	return (rand() % (r-l+1)) + l;
}
int main()
{
	srand(time(0));//初始化随机数种子
	get_rand(1,114514);//获取1->114514范围内的随机数
	return 0;
}

难点

因为我写的方法，有点和正常的不一样，更粗暴一点点
所以在合并的时候容易写错

for(int i = 0; i < tot1; i++) 
	{
		a[l + i] = b[i];
	}
	for(int i = 0; i < tot2; i++) 
	{
		a[l + tot1 + i] = c[i];
	}
	for(int i = 0; i < tot3; i++) 
	{
		a[l + tot1 + tot2 + i] = d[i];
	}

还要注意递归

quick(l, l + tot1 - 1);
quick(l + tot1 + tot2, r);