复杂度分析之leetcode: Move Zeros

今天做了一道leetcode小题,题目叫Move Zeros,地址在这里leetcode: Move Zeros

题目很简单(我就是在挑简单的题找自信),感觉是一个使用复杂度分析的好例子,简单易懂。我们一步步来,先来看一下题目。

Move Zeros

Given an arraynums, write a function to move all 0's to the end of it while maintaining the relative order of the non-zero elements.

For example, givennums = [0, 1, 0, 3, 12], after calling your function,nums should be[1, 3, 12, 0, 0].

Note:

  1. You must do this in-place without making a copy of the array.
  2. Minimize the total number of operations.

题目并不复杂,就是把一个数组里边儿的零都移到末尾。唯一需要注意的是,它要求“in-place”--就地解决,不能使用额外的存储空间,比如拷贝数组的操作是不允许的。

看完题目以后先思考几分钟吧。下面几节,我会描述一下我的思考过程。

原子操作

题目要求“in-place”的解决,那么我们这时候可以做哪些操作呢?

  • 赋值
    修改列表(数组)元素的值是可以的,这是对现有内存的操作,不会使用额外的空间。

  • 交换
    赋值可以,那么交换可以吗?我们常用的交换是三次赋值,使用一块临时空间,像这样:

     tmp = a[i]
     a[i] = a[j]
     a[j] = tmp
    

    还有不使用额外空间的技巧,像这样:

      a[i] = a[i] + a[j]
      a[j] = a[i] - a[j]
      a[i] = a[i] - a[j]
    

    所以,只要赋值操作可以做,交换就可以。

你还能想到其它的操作么?想到了一定告诉我,反正我没想到。

题目的任务现在可以理解成这样:通过赋值或者交换的手段,在不破坏非零元素顺序的情况下,把所有的零元素移到列表的末尾。

O(n^2)

我首先想到的算法是基于交换的。以题目中给出的示例为例:

    [0, 1, 0, 3, 12] => [1, 3, 12, 0, 0]

观察上面的例子,我们发现两点事实:

  1. 终结状态是确定的,一个初始状态必然对应一个确定的终结状态;
  2. 由于元素的值是不可变的,所以终结状态和初始状态的差别只是元素位置的差别。

根据以上事实,从终结状态出发可以推理出,当一个零元素的位置不正确的时候,必然占据了另一个非零元素的正确位置。

    [0, 1, 0, 3, 12] => [1, 3, 12, 0, 0] # 第一个元素“0”,占据了元素“1”的位置

如果我们能够找到那个非零元素,让两者交换位置,就可以让非零元素进入正确的位置。

    [0, 1, 0, 3, 12] => [1, 0, 0, 3, 12] # 一次交换,元素“1”进入正确的位置

由于零元素的位置不需要精确(零元素的排列顺序无意义),所以只要保证所有的非零元素的位置正确,我们就得到正确的结果了,如下

    [0, 1, 0, 3, 12] => [1, 0, 0, 3, 12] # 一次交换,元素“1”进入正确的位置
    [1, 0, 0, 3, 12] => [1, 3, 0, 0, 12] # 二次交换,元素“3”进入正确的位置
    [1, 3, 0, 0, 12] => [1, 3, 12, 0, 0] # 三次交换,元素“12”进入正确的位置

最后回答一个问题,零元素占据了那个非零元素的位置?因为非零元素的有序性,显然是从左往右最靠近它的非零元素。

根据以上分析,我们构建以下算法:

    def moveNums(nums):
        # 遍历列表
        for i in range(len(nums)):
            # 如果发现零元素,就向右寻找最靠近它的非零元素,交换两者位置
            if nums[i] == 0:
                for j in range(i+1, len(nums)):
                    if nums[j] != 0:
                        nums[i] = nums[j]
                        nums[j] = 0 
                        break

        return nums

这个算法的复杂度很好分析,两个"for-loop"的结构,显然是O(n^2)的。在leetcode上提交后的结果如下:

leetcode Submission Analysis

可以看到全部的“testcase”都通过了,但执行效率只打败了约3%的“Python Submission”。显然,这个算法的复杂度偏高了,还有更好的方法。

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

推荐阅读更多精彩内容