首先,该文章来自于极客时间网站,王争的专栏——《数据结构与算法之美》,我这里只是做简单的解释、记录并添加自己的见解,只是作为个人笔记,若侵权,马上删除。最后建议直接去该网站上购买该课程看原作者的讲解,一来是支持作者,二来是作者写的确实不错。
复杂度分析是整个算法学习的精髓,只要掌握了它,数据结构和算法的内容基本上就掌握了一半。
为什么需要复杂度分析
你可能会有些疑惑,我把代码跑一遍,通过统计、监控,就能得到算法执行的时间和占用的内存大小。为什么还要做时间、空间复杂度分析呢?这种分析方法能比我实实在在跑一遍得到的数据更准确吗?
首先,我可以肯定地说,你这种评估算法执行效率的方法是正确的。很多数据结构和算法书籍还给这种方法起了一个名字,叫事后统计法。但是,这种统计方法有非常大的局限性。
测试结果非常依赖测试环境:测试环境中硬件的不同会对测试结果有很大的影响。
测试结果受数据规模的影响很大:对同一个排序算法,待排序数据的有序度不一样,排序的执行时间就会有很大的差别。极端情况下,如果数据已经是有序的,那排序算法不需要做任何操作,执行时间就会非常短。除此之外,如果测试数据规模太小,测试结果可能无法真实地反应算法的性能。比如,对于小规模的数据排序,插入排序可能反倒会比快速排序要快!
所以,我们需要一个不用具体的测试数据来测试,就可以粗略地估计算法的执行效率的方法。这就是我们今天要讲的时间、空间复杂度分析方法。
大 O 复杂度表示法
算法的执行效率,粗略地讲,就是算法代码执行的时间。但是,如何在不运行代码的情况下,用“肉眼”得到一段代码的执行时间呢?
这里有段非常简单的代码,求1,2,3,…,n
的累加和。现在,我就带你一块来估算一下这段代码的执行时间。
1 | int cal(int n) { |
从 CPU 的角度来看,这段代码的每一行都执行着类似的操作:读数据-运算-写数据。尽管每行代码对应的 CPU 执行的个数、执行的时间都不一样,但是我们这里只是粗略估计,所以可以假设每行代码执行的时间都一样,为 unit_time。在这个假设的基础之上,这段代码的总执行时间是多少呢?
第 2、3 行代码分别需要 1 个 unit_time 的执行时间,第 4、5 行都运行了 n 遍,所以需要 $\rm 2nunit_time$ 的执行时间,所以这段代码总的执行时间就是$\rm (2n+2)unit_time$。可以看出来,所有代码的执行时间 $\rm T(n)$ 与每行代码的执行次数 n 成正比。
个人感觉最后一句话应该表述为所有代码的执行时间 $\rm T(n)$ 与所有代码的最大执行次数成正比。或者说所有代码的执行时间 $\rm T(n)$ 与问题规模 n 成正比。
按照这个分析思路,我们再来看这段代码。
1 | int cal(int n) { |
我们依旧假设每个语句的执行时间是 unit_time。那这段代码的总执行时间 T(n) 是多少呢?
第 2、3、4 行代码,每行都需要 1 个 unit_time 的执行时间,第 5、6 行代码循环执行了$\rm n$遍,需要 $\rm 2nunit_time$ 的执行时间,第 7、8 行代码循环执行了$\rm n^2$遍,所以需要$\rm 2 n^2 unit_time$的执行时间。所以,整段代码总的执行时间$\rm T(n) = (2n^2+2n+3)unit_time$。
通过这两段代码执行时间的推导过程,我们可以得到一个非常重要的规律:所有代码的执行时间 T(n) 与每行代码的执行次数 n 成正比。把这个规律总结为公式,即为大O:
其中,T(n)表示代码执行的时间;n表示数据规模的大小;$\rm f(n)$表示每行代码执行的次数总和,它是一个公式。$\rm O$表示代码的执行时间$\rm T(n)$和$\rm f(n)$表达式成正比。
所以,第一个例子中的$\rm T(n)=O(2n+2)$,第二个例子中的$\rm T(n) = O(2n^2+2n+3)$。这就是大 O 时间复杂度表示法。大 O 时间复杂度实际上并不具体表示代码真正的执行时间,而是表示代码执行时间随数据规模增长的变化趋势,所以,也叫作渐进时间复杂度(asymptotic time complexity),简称时间复杂度。
当 n 很大时,你可以把它想象成 10000、100000。而公式中的低阶、常量、系数三部分并不左右增长趋势,所以都可以忽略。我们只需要记录一个最大量级就可以了,如果用大 O 表示法表示刚讲的那两段代码的时间复杂度,就可以记为:$\rm T(n) = O(n); T(n) = O(n^2)$。
时间复杂度分析
那么如何分析一段代码的随时间复杂度呢?
只关注循环执行次数最多的一段代码
大 O 这种复杂度表示方法只是表示一种变化趋势。我们通常会忽略掉公式中的常量、低阶、系数,只需要记录一个最大阶的量级就可以了。所以,我们在分析一个算法、一段代码的时间复杂度的时候,也只关注循环执行次数最多的那一段代码就可以了。这段核心代码执行次数的 n 的量级,就是整段要分析代码的时间复杂度。
例如之前的例子:
1 | int cal(int n) { |
其中第 2、3 行代码都是常量级的执行时间,与 n 的大小无关,所以对于复杂度并没有影响。循环执行次数最多的是第 4、5 行代码,所以这块代码要重点分析。前面我们也讲过,这两行代码被执行了 n 次,所以总的时间复杂度就是 O(n)。
加法法则:总复杂度等于量级最大的那段代码的复杂度
如以下代码:
1 | int cal(int n) { |
这个代码分为三部分,分别是求 sum_1、sum_2、sum_3。我们可以分别分析每一部分的时间复杂度,然后把它们放到一块儿,再取一个量级最大的作为整段代码的复杂度。
第一段代码循环执行了 100 次,所以是一个常量的执行时间,跟 n 的规模无关(只要是一个确定的数,不管有多大,都是常量执行时间)。对于时间复杂度的概念来说,它表示的是一个算法执行效率与数据规模增长的变化趋势,所以不管常量的执行时间多大,我们都可以忽略掉,因为它本身对增长趋势并没有影响。
第二段代码的时间复杂度为$\rm O(n)$,第三段代码的时间复杂度为$\rm O(n^2)$。
综合这三段代码的时间复杂度,我们取其中最大的量级。所以,整段代码的时间复杂度就为$\rm O(n^2)$。也就是说:总的时间复杂度就等于量级最大的那段代码的时间复杂度。那我们将这个规律抽象成公式就是:
如果$\rm T1(n)=O(f(n)),T2(n)=O(g(n))$,那么:
乘法法则:嵌套代码的复杂度等于嵌套内外代码复杂度的乘积
类比上面的加法公式,乘法公式如下:
如果$\rm T1(n)=O(f(n)),T2(n)=O(g(n))$,那么:
也就是说,假设$\rm T1(n) = O(n),T2(n) = O(n^2)$,则$\rm T1(n) T2(n) = O(n^3)$。落实到具体的代码上,我们可以把乘法法则看成是*嵌套循环,例如:
1 | int cal(int n) { |
我们单独看cal()
函数。假设f()
只是一个普通的操作,那第 4~6 行的时间复杂度就是,$\rm T1(n) = O(n)$。但 f() 函数本身不是一个简单的操作,它的时间复杂度是$\rm T2(n) = O(n)$ ,所以,整个cal()
函数的时间复杂度就是,$\rm T(n) = T1(n) T2(n) = O(nn) = O(n^2)$。
几种常见时间复杂度实例分析
虽然代码千差万别,但是常见的复杂度量级并不多。稍微总结了一下,这些复杂度量级几乎涵盖了你今后可以接触的所有代码的复杂度量级。
对于刚罗列的复杂度量级,我们可以粗略地分为两类,多项式量级和非多项式量级。其中,非多项式量级只有两个:$\rm O(2^n)$ 和$\rm O(n!)$。
我们把时间复杂度为非多项式量级的算法问题叫作 NP(Non-Deterministic Polynomial,非确定多项式)问题。
当数据规模 n 越来越大时,非多项式量级算法的执行时间会急剧增加,求解问题的执行时间会无限增长。所以,非多项式时间复杂度的算法其实是非常低效的算法。因此,关于 NP 时间复杂度就不展开讲了。我们主要来看几种常见的多项式时间复杂度。
O(1)
O(1) 只是常量级时间复杂度的一种表示方法,并不是指只执行了一行代码。比如这段代码,即便有 3 行,它的时间复杂度也是 O(1),而不是 O(3)。
1 | int i = 8; |
稍微总结一下,只要代码的执行时间不随 n 的增大而增长,这样代码的时间复杂度我们都记作 O(1)。或者说,一般情况下,只要算法中不存在循环语句、递归语句,即使有成千上万行的代码,其时间复杂度也是Ο(1)。
O(logn)、O(nlogn)
对数阶时间复杂度非常常见,同时也是最难分析的一种时间复杂度。通过一个例子来说明一下。
1 | i=1; |
根据我们前面讲的复杂度分析方法,第三行代码是循环执行次数最多的。所以,我们只要能计算出这行代码被执行了多少次,就能知道整段代码的时间复杂度。
从代码中可以看出,变量 i 的值从 1 开始取,每循环一次就乘以 2。当大于 n 时,循环结束。实际上,变量$\rm i$的取值就是一个等比数列。如果一个个列出来,应该是这样的
所以,我们只要知道 x 值是多少,就知道这行代码执行的次数了。通过$\rm 2^x=n$求解 x 这个问题得到结果:$\rm x=log_2n$,所以,这段代码的时间复杂度就是$\rm O(log_2n)$。
把代码稍微改下,如下所示,此时代码的时间复杂度就是$\rm O(log_3n)$。
1 | i=1; |
实际上,不管是以 2 为底、以 3 为底,还是以 10 为底,我们可以把所有对数阶的时间复杂度都记为 O(logn)。
我们知道,对数之间是可以互相转换的,$\rm log_3n$就等于$\rm log_32 log_2n$,所以$\rm O(log_3n) = O(C log_2n)$,其中 $\rm C=log_32$是一个常量。基于我们前面的一个理论:在采用大 O 标记复杂度的时候,可以忽略系数,即$\rm O(Cf(n)) = O(f(n))$。所以,就$\rm O(log_2n)$就等于$\rm O(log_3n)$。因此,在对数阶时间复杂度的表示方法里,我们忽略对数的“底”,统一表示为$\rm O(logn)$。
如果你理解了我前面讲的$\rm O(logn)$,那 $\rm O(nlogn)$ 就很容易理解了。还记得我们刚讲的乘法法则吗?如果一段代码的时间复杂度是 $\rm O(logn)$,我们循环执行 n 遍,时间复杂度就是 $\rm O(nlogn)$ 了。而且,$\rm O(nlogn)$也是一种非常常见的算法时间复杂度。比如,归并排序、快速排序的时间复杂度都是 $\rm O(nlogn)$。
O(m+n)、O(m*n)
我们再来看一种跟前面都不一样的时间复杂度,代码的复杂度由两个数据的规模来决定。
1 | int cal(int m, int n) { |
从代码中可以看出,m 和 n 是表示两个数据规模。我们无法事先评估 m 和 n 谁的量级大,所以我们在表示复杂度的时候,就不能简单地利用加法法则,省略掉其中一个。所以,上面代码的时间复杂度就是 O(m+n)。
针对这种情况,原来的加法法则就不正确了,我们需要将加法规则改为$\rm T1(m) + T2(n) = O(f(m) + g(n))$。但是乘法法则继续有效:$\rm T1(m)T2(n) = O(f(m) f(n))$。
空间复杂度分析
时间复杂度的全称是渐进时间复杂度,表示算法的执行时间与数据规模之间的增长关系。类比一下,空间复杂度全称就是渐进空间复杂度(asymptotic space complexity),表示算法的存储空间与数据规模之间的增长关系。
例如下面代码:
1 | void print(int n) { |
跟时间复杂度分析一样,我们可以看到,第 2 行代码中,我们申请了一个空间存储变量 i,但是它是常量阶的,跟数据规模 n 没有关系,所以我们可以忽略。第 3 行申请了一个大小为 n 的 int 类型数组,除此之外,剩下的代码都没有占用更多的空间,所以整段代码的空间复杂度就是 O(n)。
我们常见的空间复杂度就是 O(1)、O(n)、O(n2),像 O(logn)、O(nlogn) 这样的对数阶复杂度平时都用不到。而且,空间复杂度分析比时间复杂度分析要简单很多。因此掌握上面所说内容已经足够了。
内容小结
复杂度也叫渐进复杂度,包括时间复杂度和空间复杂度,用来分析算法执行效率与数据规模之间的增长关系,可以粗略地表示,越高阶复杂度的算法,执行效率越低。常见的复杂度并不多,从低阶到高阶有:O(1)、O(logn)、O(n)、O(nlogn)、$\rm O(n^2)$。