LeetCode 力扣 75. 颜色分类

题目描述(中等难度)

给一个数组,含有的数只可能 0,1,2 中的一个,然后把这些数字从小到大排序。

解法一

题目下边的 Follow up 提到了一个解法,遍历一次数组,统计 0 出现的次数,1 出现的次数,2 出现的次数,然后再遍历数组,根据次数,把数组的元素改成相应的值。当然我们只需要记录 0 的次数,和 1 的次数,剩下的就是 2 的次数了。

public void sortColors(int[] nums) {
    int zero_count = 0;
    int one_count = 0;
    for (int i = 0; i < nums.length; i++) {
        if (nums[i] == 0) {
            zero_count++;
        }
        if (nums[i] == 1) {
            one_count++;
        }
    }
    for (int i = 0; i < nums.length; i++) {
        if (zero_count > 0) {
            nums[i] = 0;
            zero_count--;
        } else if (one_count > 0) {
            nums[i] = 1;
            one_count--;
        } else {
            nums[i] = 2;
        }
    }
}

时间复杂度:O(n)。

空间复杂度:O(1)。

解法二

上边的算法,我们遍历了两次数组,让我们想一想只遍历一次的方法。我们假设一种简单的情况,如果只含有两个数 0 和 1,该怎么做呢?

假设原数组是 1 0 1 1 0,我们可以用一个指针,zero_position,含义是该指针指向的位置,前边的位置全部存 0 。然后再用一个指针 i 遍历这个数组,找到 0 就把 0 放到当前 zero_position 指向的位置, zero_position 后移。用 Z 代表 zero_position,看下边的遍历过程。

1 0 1 1 0   初始化 Z,i 指向第 0 个位置,i 后移
^
Z
i

1 0 1 1 0   发现 0,把 Z 的位置置为 0,并且把 Z 的位置的数字交换过来,Z 后移一位
^ ^
Z i

0 1 1 1 0   i 后移一位
  ^
  i
  Z
  
0 1 1 1 0  i 继续后移
  ^ ^
  Z i

0 1 1 1 0  i 继续后移
  ^   ^
  Z   i
  
0 1 1 1 0  遇到 0,把 Z 的位置置为 0,并且把 Z 的位置的数字交换过来,Z 后移一位
  ^     ^
  Z     i
  
0 0 1 1 1  遍历结束
    ^   ^
    Z   i

回到我们当前这道题,我们有 3 个数字,那我们可以用两个指针,一个是 zero_position,和之前一样,它前边的位置全部存 0。再来一个指针,two_position,注意这里是,它后边的位置全部存 2。然后遍历整个数组就行了。

下边画一个遍历过程中的图,理解一下,Z 前边全存 0,T 后边全存 2。

0 1 0 2 1 2 2 2
  ^ ^   ^
  Z i   T
public void sortColors(int[] nums) {
    int zero_position = 0;
    int two_position = nums.length - 1;
    for (int i = 0; i <= two_position; i++) {
        if (nums[i] == 0) {
            //将当前位置的数字保存
            int temp = nums[zero_position];
            //把 0 存过来
            nums[zero_position] = 0;
            //把之前的数换过来
            nums[i] = temp;
            //当前指针后移
            zero_position++;
        } else if (nums[i] == 2) {
            //将当前位置的数字保存
            int temp = nums[two_position];
            //把 2 存过来
            nums[two_position] = 2;
            //把之前的数换过来
            nums[i] = temp;
            //当前指针前移
            two_position--;
            //这里一定要注意,因为我们把后边的数字换到了第 i 个位置,
            //这个数字我们还没有判断它是多少,外层的 for 循环会使得 i++ 导致跳过这个元素
            //所以要 i--
            //而对于上边 zero_position 的更新不需要考虑,因为它是从前边换过来的数字
            //在之前已经都判断过了
            i--;
        }
    }
}

时间复杂度:O(n)。

空间复杂度:O(1)。

解法三

解法二中总共有三种数,然后很自然可以分成三部分,用两个指针作为间隔,但是,如果有 5 种数呢,解法二恐怕就不适用了。在 leetcode 发现另一种解法,参考这里的解法二,用了大问题化小的思想。

我们用三个指针 n0,n1,n2,分别代表已排好序的数组当前 0 的末尾,1 的末尾,2 的末尾。

0  0  1  2  2  2  0  2  1
   ^  ^        ^  ^
  n0 n1       n2  i

然后当前遍历到 i 的位置,等于 0,我们只需要把 n2 指针后移并且将当前数字置为 2,将 n1 指针后移并且将当前数字置为 1,将 n0 指针后移并且将当前数字置为 0。

0  0  1  2  2  2  2  2  1  n2 后移后的情况 
   ^  ^           ^  
   n0 n1          i
                  n2  
                  
0  0  1  1  2  2  2  2  1  n1 后移后的情况
   ^     ^        ^  
   n0    n1       i
                  n2                   

0  0  0  1  2  2  2  2  1  n0 后移后的情况
      ^  ^        ^  
      n0 n1       i
                  n2                    

然后就达到了将 i 指向的 0 插入到当前排好序的 0 的位置的末尾。

原因的话,由于前边插入了新的数字,势必造成数字的覆盖,指针后移后要把对应的指针位置置为对应的数,n2 指针后移后置为 2,n1 指针后移后置为 1,例如,假如之前有 3 个 2,由于前边插入一个数字,所以会导致 1 个 2 被覆盖掉,所以要加 1 个 2。

public void sortColors(int[] nums) {
    int n0 = -1, n1 = -1, n2 = -1;
    int n = nums.length;
    for (int i = 0; i < n; i++) {
        if (nums[i] == 0) {
            n2++;
            nums[n2] = 2;
            n1++;
            nums[n1] = 1;
            n0++;
            nums[n0] = 0;
        } else if (nums[i] == 1) {
            n2++;
            nums[n2] = 2;
            n1++;
            nums[n1] = 1;
        } else if (nums[i] == 2) {
            n2++;
            nums[n2] = 2;
        }
    }
}

时间复杂度:O(n)。

空间复杂度:O(1)。

解法二利用指针,在原来的空间存东西很经典。解法三,其实本质是我们常用的递归思想,先假设一个小问题解决了,然后假如再来一个数该怎么操作。

更多详细通俗题解详见 leetcode.wang

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,634评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,951评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,427评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,770评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,835评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,799评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,768评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,544评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,979评论 1 308
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,271评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,427评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,121评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,756评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,375评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,579评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,410评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,315评论 2 352

推荐阅读更多精彩内容

  • 题目描述(困难难度) 黑色的看成墙,蓝色的看成水,宽度一样,给定一个数组,每个数代表从左到右墙的高度,求出能装多少...
    windliang阅读 562评论 0 0
  • 数组中重复的数字(一维数组) 问题:在一个长度为长度为n的数组里的所有数字都在0-n-1之间。找出任意一个重复的数...
    Drama_Du阅读 1,120评论 0 0
  • 1. 找出数组中重复的数字 题目:在一个长度为n的数组里的所有数字都在0到n-1的范围内。数组中某些数字是重复的,...
    BookThief阅读 1,754评论 0 2
  • 1. 链表 链表是最基本的数据结构,面试官也常常用链表来考察面试者的基本能力,而且链表相关的操作相对而言比较简单,...
    Mr希灵阅读 1,439评论 0 20
  • 一、框架 1、Mac系统及常用工具、进制;C数据类型、常量变量、运算符、表达式、格式化输入输出 2、关系运算符、逻...
    师景福阅读 687评论 0 1