常用的排序算法------练习4

日暮南城故里2025-03-31 11:53

1. 题目

2. 思路和题解

这道题是很经典的荷兰国旗问题，根据题目意思，要对这个数组按照颜色排序，而此时现在的红、白、蓝三个颜色分别对应0，1，2，因此可以想到使用冒泡排序对该数组进行排序。

代码如下：

java 复制代码

class Solution {
    public void sortColors(int[] nums) {
        for (int i = 0; i < nums.length; i++) {
            for (int j = nums.length - 1; j > i; j--) {
                if (nums[j - 1] > nums[j]) {
                    int temp = nums[j];
                    nums[j] = nums[j - 1];
                    nums[j - 1] = temp;
                }
            }
        }
    }
}

虽然这种方法可以通过，但是时间复杂度很高，然后查看了官方给出的题解，官方是先统计数组中0，1，2的个数，然后根据他们的数量，重写整个数组。初始化两个指针分别指向0和nums.length - 1，然后如果遇到0，就交换到数组的头部，遇到2，就交换到数组的尾部，当遍历的数组超过了右指针，则遍历结束。这一需要注意的一点是，当找到2时，需要不断地将其进行交换，直到新的nums[i]不为2，才能停止交换。

代码如下：

java 复制代码

class Solution {
    public void sortColors(int[] nums) {
        int left = 0, right = nums.length - 1;
        for (int i = 0; i <= right; ++i) {
            while (i <= right && nums[i] == 2) {
                int temp = nums[i];
                nums[i] = nums[right];
                nums[right] = temp;
                --right;
            }
            if (nums[i] == 0) {
                int temp = nums[i];
                nums[i] = nums[left];
                nums[left] = temp;
                ++left;
            }
        }
    }
}

用这种方法，时间复杂度就低很多了，也能更适用于普遍的情况。