Q6.特定分治策略

“6.5.7选第k小元素：特定分治策略”
在给出的一个数组S中，找到第k小的数。

解析

以S中的某个元素m作为划分标准，将S划分为两个子数组S1和S2，把这个数组中比m小的都放入S1的数组中，数组S1的元素个数是|S1|个；把这个数组中比m*大的都放入S2的数组中，数组S2的元素个数是|S2|个。

若k<|S1|，则原问题归纳为在数组S1中找第k小的子问题。
若k=|S1|+1，则m*就是要找的第k小元素。
若k>|S1|+1，则原问题归纳为在数组S2中找第n−|S1|−1小的子问题。

设计

输入: n个数的数组S，正整数k
输出: S中的第k小元素
1.将S划分成5个一组,共「n/5」个组。
2.每组找一个中位数，把这些中位数放到集合M中
3. m←Select(M,「|M|/27」 //选M的中位数m，将S中的数划分成A,B,C,D四个集合
4.把A和D中的每个元素与m比较,小的构成S1,大的构成S2 ;
5. S1←S1∪C；S2←S2∪B
6. if k=|S1|+1 then 输出m
7. else if k≤|S1 |
8.then Select(S1, k)
9.else Select(S2, k-|S1|-1)

int Select(int *a, int len, int k)
{
	int cnt = len / 5;
	int m[MAX], num = 0;
	int s1[MAX], s2[MAX];
	int num_s1 = 0, num_s2 = 0;

	if (len <= 5)
	{
		qsort(a, len, sizeof(int), comp);
		return a[k - 1];
	}
	for (int i = 0; i < cnt; i++)
	{

		qsort(a + 5 * i, 5, sizeof(int), comp);
		m[num++] = a[5 * i + 2];
	}

	int m_s = Select(m, num, num / 2 + 1);

	for (int i = 0; i < cnt; i++)
	{
		if (a[5 * i + 2] < m_s)
		{
			s1[num_s1++] = a[5 * i];
			s1[num_s1++] = a[5 * i + 1];
			s1[num_s1++] = a[5 * i + 2];

			if (a[5 * i + 3] < m_s)
				s1[num_s1++] = a[5 * i + 3];
			else
				s2[num_s2++] = a[5 * i + 3];

			if (a[5 * i + 4] < m_s)
				s1[num_s1++] = a[5 * i + 4];
			else
				s2[num_s2++] = a[5 * i + 4];
		}
		else if (a[5 * i + 2] == m_s)
		{
			s1[num_s1++] = a[5 * i];
			s1[num_s1++] = a[5 * i + 1];
			s2[num_s2++] = a[5 * i + 3];
			s2[num_s2++] = a[5 * i + 4];
		}
		else
		{
			s2[num_s2++] = a[5 * i + 3];
			s2[num_s2++] = a[5 * i + 4];
			s2[num_s2++] = a[5 * i + 2];

			if (a[5 * i] < m_s)
				s1[num_s1++] = a[5 * i];
			else
				s2[num_s2++] = a[5 * i];

			if (a[5 * i + 1] < m_s)
				s1[num_s1++] = a[5 * i + 1];
			else
				s2[num_s2++] = a[5 * i + 1];
		}
	}

	for (int i = 5 * cnt; i < len; i++)
	{
		if (a[i] < m_s)
			s1[num_s1++] = a[i];
		else
			s2[num_s2++] = a[i];
	}

	if (k == num_s1 + 1)
		return m_s;
	else if (k <= num_s1)
		return Select(s1, num_s1, k);
	else
		return Select(s2, num_s2, k - num_s1 - 1);
}

分析

时间复杂度是O(n)

源码

https://github.com/pt0918/-/blob/master/Specific_division.cpp