快速排序算法改进:随机快排-荷兰国旗划分详解-CSDN博客

本文链接：https://blog.csdn.net/weixin_45562399/article/details/148542814

随机快速排序-荷兰国旗划分算法详解

快速排序是经典的平均时间复杂度 $O (n)$ 的排序算法，而随机快排结合荷兰国旗划分（Dutch National Flag Partitioning）的优化策略，进一步提升了性能和稳定性。本文我将深入剖析随机快排 - 荷兰国旗划分的原理、实现细节、性能分析以及实际应用场景，带你全面掌握这一强大的排序技术。

一、基础知识回顾

1.1 快速排序简介

快速排序是一种基于分治思想的排序算法。基本思路是：从数组中选择一个元素作为枢轴（pivot），将数组分为两部分，使得左边部分的元素都小于等于枢轴，右边部分的元素都大于枢轴，然后递归地对左右两部分进行排序，最终实现整个数组的有序排列。快速排序在平均情况下的时间复杂度为 $O (n l o g n)$ ，但在最坏情况下（如数组已经有序且每次选择第一个或最后一个元素作为枢轴），时间复杂度会退化为 $O(n^2)$ 。

1.2 荷兰国旗问题

荷兰国旗由红、白、蓝三种颜色的条纹组成，荷兰国旗问题是指：给定一个由红色、白色和蓝色三种颜色的小球组成的数组，要求将数组中的元素按照红、白、蓝的顺序排列，且尽可能地提高排序效率。在算法实现中，用0表示红色，1表示白色，2表示蓝色。荷兰国旗划分算法通过一次遍历，将数组划分为小于枢轴、等于枢轴和大于枢轴的三个区域，这种划分方式为随机快排的优化提供了思路。

二、随机快排 - 荷兰国旗划分原理

2.1 随机化枢轴选择

为了避免快速排序在最坏情况下时间复杂度退化，随机快排采用随机选择枢轴的策略，在每次进行划分操作前，从待排序数组中随机选择一个元素作为枢轴。这样做使得无论输入数组的初始顺序如何，都能在大概率下保证划分的均衡性，算法的平均时间复杂度从而稳定在 $O (n l o g n)$ 。

2.2 荷兰国旗划分过程

荷兰国旗划分的核心在于将数组划分为三个区间：

小于枢轴区间：位于数组的左侧，包含所有小于枢轴的元素。
等于枢轴区间：位于小于枢轴区间和大于枢轴区间之间，包含所有等于枢轴的元素。
大于枢轴区间：位于数组的右侧，包含所有大于枢轴的元素。

在划分过程中，使用三个指针：

left指针：指向小于枢轴区间的右边界，初始位置为数组起始位置 - 1。
current指针：用于遍历数组，从数组起始位置开始。
right指针：指向大于枢轴区间的左边界，初始位置为数组末尾位置。

划分过程如下：

当 current 指针小于等于 right 指针时，进行循环：
- 若 arr[current] 小于枢轴，left 指针右移一位，交换 arr[left] 和 arr[current]，然后 current 指针右移一位。
- 若 arr[current] 等于枢轴，current 指针直接右移一位。
- 若 arr[current] 大于枢轴，交换 arr[current] 和 arr[right]，right 指针左移一位。
循环结束后，数组被成功划分为三个区间，小于枢轴的元素在左侧，等于枢轴的元素在中间，大于枢轴的元素在右侧。

2.3 结合随机快排与荷兰国旗划分

在随机快排中引入荷兰国旗划分后，排序过程如下：

随机选择一个枢轴元素。
使用荷兰国旗划分算法将数组划分为三个区间：小于枢轴区间、等于枢轴区间和大于枢轴区间。
递归地对小于枢轴区间和大于枢轴区间进行随机快排。
由于等于枢轴区间的元素已经有序，无需再次排序，最终得到整个有序数组。

通过这种方式，随机快排 - 荷兰国旗划分不仅避免了传统快排的最坏情况，还能在一次划分中处理所有等于枢轴的元素，减少了递归调用的次数，进一步提高了排序效率。

三、代码实现

3.1 Python实现

import random


def dutch_national_flag_partition(arr, pivot_index):
    pivot = arr[pivot_index]
    left, current, right = 0, 0, len(arr) - 1
    while current <= right:
        if arr[current] < pivot:
            arr[left], arr[current] = arr[current], arr[left]
            left += 1
            current += 1
        elif arr[current] == pivot:
            current += 1
        else:
            arr[current], arr[right] = arr[right], arr[current]
            right -= 1
    return left, right


def randomized_quicksort(arr):
    if len(arr) <= 1:
        return arr
    pivot_index = random.randint(0, len(arr) - 1)
    left, right = dutch_national_flag_partition(arr, pivot_index)
    left_part = randomized_quicksort(arr[:left])
    right_part = randomized_quicksort(arr[right + 1:])
    equal_part = arr[left:right + 1]
    return left_part + equal_part + right_part


# 测试
arr = [2, 0, 2, 1, 1, 0]
print(randomized_quicksort(arr))

3.2 Java实现

import java.util.Arrays;
import java.util.Random;

public class RandomizedQuicksort {
    public static int[] dutchNationalFlagPartition(int[] arr, int pivotIndex) {
        int pivot = arr[pivotIndex];
        int left = 0, current = 0, right = arr.length - 1;
        while (current <= right) {
            if (arr[current] < pivot) {
                swap(arr, left, current);
                left++;
                current++;
            } else if (arr[current] == pivot) {
                current++;
            } else {
                swap(arr, current, right);
                right--;
            }
        }
        return new int[]{left, right};
    }

    public static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

    public static int[] randomizedQuicksort(int[] arr) {
        if (arr.length <= 1) {
            return arr;
        }
        Random random = new Random();
        int pivotIndex = random.nextInt(arr.length);
        int[] partitionIndices = dutchNationalFlagPartition(arr, pivotIndex);
        int left = partitionIndices[0];
        int right = partitionIndices[1];
        int[] leftPart = randomizedQuicksort(Arrays.copyOfRange(arr, 0, left));
        int[] rightPart = randomizedQuicksort(Arrays.copyOfRange(arr, right + 1, arr.length));
        int[] equalPart = Arrays.copyOfRange(arr, left, right + 1);
        int[] result = new int[leftPart.length + equalPart.length + rightPart.length];
        System.arraycopy(leftPart, 0, result, 0, leftPart.length);
        System.arraycopy(equalPart, 0, result, leftPart.length, equalPart.length);
        System.arraycopy(rightPart, 0, result, leftPart.length + equalPart.length, rightPart.length);
        return result;
    }

    public static void main(String[] args) {
        int[] arr = {2, 0, 2, 1, 1, 0};
        System.out.println(Arrays.toString(randomizedQuicksort(arr)));
    }
}

3.3 C++实现

#include <iostream>
#include <vector>
#include <ctime>
#include <cstdlib>
using namespace std;

pair<int, int> dutch_national_flag_partition(vector<int>& arr, int pivot_index) {
    int pivot = arr[pivot_index];
    int left = 0, current = 0, right = arr.size() - 1;
    while (current <= right) {
        if (arr[current] < pivot) {
            swap(arr[left], arr[current]);
            left++;
            current++;
        }
        else if (arr[current] == pivot) {
            current++;
        }
        else {
            swap(arr[current], arr[right]);
            right--;
        }
    }
    return make_pair(left, right);
}

vector<int> randomized_quicksort(vector<int> arr) {
    if (arr.size() <= 1) {
        return arr;
    }
    srand(time(nullptr));
    int pivot_index = rand() % arr.size();
    pair<int, int> partition_indices = dutch_national_flag_partition(arr, pivot_index);
    int left = partition_indices.first;
    int right = partition_indices.second;
    vector<int> left_part = randomized_quicksort(vector<int>(arr.begin(), arr.begin() + left));
    vector<int> right_part = randomized_quicksort(vector<int>(arr.begin() + right + 1, arr.end()));
    vector<int> equal_part(arr.begin() + left, arr.begin() + right + 1);
    vector<int> result;
    result.reserve(left_part.size() + equal_part.size() + right_part.size());
    result.insert(result.end(), left_part.begin(), left_part.end());
    result.insert(result.end(), equal_part.begin(), equal_part.end());
    result.insert(result.end(), right_part.begin(), right_part.end());
    return result;
}

int main() {
    vector<int> arr = {2, 0, 2, 1, 1, 0};
    vector<int> sorted_arr = randomized_quicksort(arr);
    for (int num : sorted_arr) {
        cout << num << " ";
    }
    cout << endl;
    return 0;
}

四、性能分析

4.1 时间复杂度

随机快排 - 荷兰国旗划分在平均情况下的时间复杂度为 $O (n l o g n)$ 。由于随机选择枢轴，使得每次划分都能大概率地将数组分成两个大致相等的子数组，递归深度为 $O (l o g n)$ ，每次划分操作的时间复杂度为 $O (n)$ ，因此整体时间复杂度为 $O (n l o g n)$ 。在最坏情况下，虽然随机化枢轴选择降低了出现的概率，但仍有可能每次都选择到数组中的最大值或最小值，导致划分不均衡，此时时间复杂度退化为 $O(n^2)$ 。不过，这种极端情况发生的概率极小。

4.2 空间复杂度

随机快排 - 荷兰国旗划分的空间复杂度主要取决于递归调用栈的深度。在平均情况下，递归深度为 $O (l o g n)$ ，因此空间复杂度为 $O (l o g n)$ ；在最坏情况下，递归深度为 $O (n)$ ，空间复杂度为 $O (n)$ 。此外，在划分过程中，虽然没有使用额外的与数组长度成正比的空间，但存在一些临时变量（如指针和交换过程中的临时存储），这些变量占用的空间为常数级，对整体空间复杂度影响较小。