首页 > 其他 > 详细

颜色分类

时间:2019-08-25 00:48:32      阅读:164      评论:0      收藏:0      [点我收藏+]

题目链接:

涉及知识:数组、双指针

问题描述:

? 该问题又被称为荷兰国旗问题,大概意思为给每一个数字指定一种颜色,并按荷兰国旗颜色的顺序进行调整。

算法思想:

? 设置三个指针,初始状态 low 指针指向最低位, high 指针指向高位,并设置 cur 指针指向当前的元素。cur 指针从头开始遍历。

  • 初始化 0 的最右边界:low = 0。

  • 初始化 2 的最左边界 :high = n - 1。

  • 初始化当前考虑的元素序号 :cur = 0.
    • While cur <= high :
    • 若 nums[cur] = 0 :交换第 cur 个和第 low 个元素,并将指针都向右移。
    • 若 nums[cur] = 2 :交换第 cur 个和第 high 个元素,并将 high 指针左移 。
    • 若 nums[cur] = 1 :将指针 cur 右移。

为什么和低位交换之后,两个指针均向右移,而和高位交换之后只将 high 指针向左移动?

? 因为和低位交换之后,cur 所指向的元素为 1(原先的 low 指针指向的为 1,当然刚开始的时候可能为 0,但是没有影响),而 high 指针原先指向的元素可能为 2,交换之后 cur 指向的元素可能为 2,需要在和 high 指向的元素进行交换,因此此时不可以将 cur 向右移动。

代码:

class Solution {
    public void sortColors(int[] nums) {
        int low = 0;
        int high = nums.length - 1;
        int cur = 0;

        while(cur <= high){
            if(nums[cur] == 0){
                int tmp = nums[cur];
                nums[cur] = nums[low];
                nums[low] = tmp;
                low++;
                cur++;
            }else if(nums[cur] == 2) {
                int tmp = nums[cur];
                nums[cur] = nums[high];
                nums[high] = tmp;
                high--;
            } else {
                cur++;
            }
        }

        return;
    }
}

颜色分类

原文:https://www.cnblogs.com/zcxhaha/p/11406643.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!