快速排序中的堆栈深度

快速排序的伪代码如下:

1 QUICKSORT(A, p, r)
2     if p < r
3         q <- PARTITION(A, p, r)
4         QUICISORT(A, p, q-1)
5         QUICKSORT(A, q + 1, r)

可采用尾递归的方式减小堆栈的深度,即采用迭代控制结构替代第二次递归调用,尾递归在大多数的编译程序中都被采用。伪代码如下:

1 QUICKSORT(A, p, r)
2     while p < r
3         q <- PARTITION(A, p, r)
4         QUICISORT(A, p, q-1)
5         p <- q + 1

尽管此时平均堆栈深度已减小,但最坏情况下堆栈深度仍为θ(n),如数组元素已有序的情况下进行快速排序。那如何减小最坏情况下的堆栈深度?
核心思想仍然是采用迭代控制结构替代递归调用。前一次的思想是执行PARTITION后,对数组的前一部分( A[p...q-1] )进行递归排序,对后一部分( A[q+1...r] )进行迭代控制运算。
假设每次执行PARTITION后,(q-1) - p + 1 : r-(q+1) +1 = a ( a > 0 ),则有

n * a ^ h = 1
=> h = - lgn / lga
  • a < 1/2 => h < lgn
  • a = 1/2 => h = lgn
  • a > 1/2 => h > lgn

因此,执行完PARTITION后,对元素个数少的那一部分进行递归排序,而对另一部分进行迭代运算,可使堆栈深度小于lgn。

伪代码如下:

1 QUICKSORT(A, p, r)
2     while p < r
3         q <- PARTITION(A, p, r)
4         if (q - 1) - p + 1 < r - (q + 1) + 1
5             QUICISORT(A, p, q-1)
6             p = q + 1
7         else  
8             QUICKSORT(A, q + 1, r)
9             r = q - 1
posted @ 2012-11-07 10:02  zzwab  阅读(636)  评论(0编辑  收藏  举报