让我们从用 C 重写您的代码开始,因为 C 对我来说更容易解释。因此,让我们用一些注释来回忆您的代码:
int
counting_sort(int a[], int a_len, int maxVal)
{
int i, j, outPos = 0;
int bucket_len = maxVal+1;
int bucket[bucket_len]; /* simple bucket structure */
memset(bucket, 0, sizeof(int) * bucket_len);
/* one loop bucket processing */
for (i = 0; i < a_len; i++)
{
bucket[a[i]]++; /* simple work with buckets */
}
for (i=0; i < bucket_len; i++)
{
for (j = 0; j < bucket[i]; j++)
{
a[outPos++] = i;
}
}
return 0;
}
现在让我们向这个人提供一些真实的数据:
[126、348、343、432、316、171、556、223、670、201]
在输出上我们有
[126、171、201、223、316、343、348、432、556、670]
似乎一切正常?还没有。让我们看看maxVal。它是 670(!)为了在这里对 10 个元素的数组进行排序,我们使用了 670 个元素的数组,主要是零。非常。为了处理这个计数排序问题,我们有两种可能的泛化方式:
1)第一种方式——按数字排序。这称为基数排序。让我们展示一些代码,尝试使其尽可能接近计数排序代码。再看评论:
int
radix_sort(int a[], int a_len, int ndigits)
{
int i;
int b[a_len];
int expn = 1;
/* additional loop for digits */
for (i = 0; i != ndigits; ++i)
{
int j;
int bucket[10] = {0}; /* still simple buckets */
/* bucket processing becomes tricky */
for (j = 0; j != a_len; ++j)
bucket[ a[j] / expn % 10 ]++;
for (j = 1; j != 10; ++j)
bucket[j] += bucket[j - 1];
for (j = a_len - 1; j >= 0; --j)
b[--bucket[a[j] / expn % 10]] = a[j];
for (j = 0; j != a_len; ++j)
a[j] = b[j];
expn *= 10;
}
}
我们用 N 附近的乘数换取内存。利润?也许。但在某些情况下,N 附近的乘数非常重要。程序、一天工作和一周工作与用户视图有很大不同,即使两者分别工作 1*O(N) 和 7*O(N)。所以我们要进行第二次概括:
2)第二种方式——让桶更复杂。这称为桶排序。
让我们再次从一些代码开始。在哲学论证之前,我更喜欢更多的代码。还是看评论吧,都是必不可少的。
int
bucket_sort(int a[], int a_len, int maxVal)
{
int i, aidx;
typedef struct tag_list {
int elem;
struct tag_list *next;
} list_t, *list_p;
list_p bucket[10] = {0}; /* sophisticated buckets */
/* one loop simple processing with one more inner loop
to get sorted buckets (insert-sort on lists, Cormen-style) */
for (i = 0; i != a_len; ++i)
{
int bnum = (10 * a[i]) / maxVal;
list_p bptr = bucket[bnum];
list_p belem = malloc(sizeof(list_t));
belem->elem = a[i];
if (bptr == 0)
{
bucket[bnum] = belem;
belem->next = 0;
continue;
}
else if (a[i] <= bptr->elem)
{
belem->next = bptr;
bucket[bnum] = belem;
continue;
}
else
{
while (bptr != 0)
{
if ((bptr->elem <= a[i]) && ((bptr->next == 0) || (bptr->next->elem > a[i])))
{
belem->next = bptr->next;
bptr->next = belem;
break;
}
bptr = bptr->next;
}
}
}
/* one loop (looks as two) to get all back */
aidx = 0;
for (i = 0; i != 10; ++i)
{
list_p bptr = bucket[i];
while (bptr)
{
list_p optr = bptr;
a[aidx] = bptr->elem;
aidx += 1;
bptr = bptr->next;
free(optr);
}
}
return 0;
}
那么我们这里有什么?我们正在交易一些复杂的存储桶结构和动态分配内存的要求,但赢得了静态内存和平均接近 N 的乘数。
现在让我们回忆一下我们在代码中看到的内容:
- 计数排序——简单的桶、简单的处理、内存开销
- 基数排序——简单的桶、复杂的处理、速度开销(并且仍然需要额外的静态内存)
- 桶排序——复杂的桶,处理简单,需要动态内存,平均不错
因此,基数和桶排序是计数排序的两个有用的概括。它们与计数排序和彼此有很多共同点,但在每种情况下,我们都在失去一些东西并赢得一些东西。软件工程是关于这些机会之间的平衡。