给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

    此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

    注意:
    不能使用代码库中的排序函数来解决这道题。

    示例:

    输入: [2,0,2,1,1,0]
    输出: [0,0,1,1,2,2]
    进阶:

    一个直观的解决方案是使用计数排序的两趟扫描算法。
    首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
    你能想出一个仅使用常数空间的一趟扫描算法吗?

    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/sort-colors
    著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

    1. class Solution {
    2. public void sortColors(int[] nums) {
    3. // 双指针
    4. int low = 0,high = nums.length - 1;
    5. int cur = 0;
    6. int temp;
    7. while(cur <= high){
    8. if(nums[cur] == 0){
    9. temp = nums[low];
    10. nums[low] = nums[cur];
    11. nums[cur] = temp;
    12. low ++;
    13. cur ++;
    14. }
    15. else if(nums[cur] == 2){
    16. temp = nums[high];
    17. nums[high] = nums[cur];
    18. nums[cur] = temp;
    19. high --;
    20. }
    21. else cur ++;
    22. }
    23. }
    24. }

    时间复杂度:O(N)
    空间复杂度:O(1)