“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