本文共 1030 字，大约阅读时间需要 3 分钟。

要解决给定的问题，我们需要对数组进行原地排序，使得相同颜色的元素相邻，并按照红色（0）、白色（1）、蓝色（2）的顺序排列。以下是详细的步骤说明：

解题思路

这种方法时间复杂度为 O(n)，其中 n 是数组的长度。因为我们只需要两次遍历（一次统计，一次重写）即可完成任务，因此效率很高。

代码实现

public class Solution {    public void sortColors(int[] nums) {        int count0 = 0, count1 = 0, count2 = 0;        for (int num : nums) {            if (num == 0) {                count0++;            } else if (num == 1) {                count1++;            } else {                count2++;            }        }        int index = 0;        while (index < nums.length) {            for (int i = 0; i < count0; i++) {                nums[index++] = 0;            }            for (int i = 0; i < count1; i++) {                nums[index++] = 1;            }            for (int i = 0; i < count2; i++) {                nums[index++] = 2;            }        }    }}

解释