3 sum problem问题: given an array, you have to find if there exists 3 numbers in the array, a, b, c such that a + b + c = 0.
例如:
a1, a2, ....., an, 我们需要从这组数字中找到三个数字:
ap, am, aq , such that:
ap + am + aq = 0。
具体的, 如果我们的数组中数字如下:
10, -8, -2, 5, -3, 4
如何从中找到三个数a, b, c 满足a + b + c = 0呢?
这是一个很简单的例子, 可以直接观察有:
10 + (-8) + (-2) = 0 or (-2) + 5 + (-3) = 0。
上述的问题就是3 sum problem。 是sum of subset problem 的特殊形式。
解决办法:
(1) brute force 搜索, 需要检查Cn^3 = n (n -1) (n -2) /6 = O(n^3)中可能的组合形式。 (n 是数组中存放的总的数字的个数)。
假如我们的数组的长度是N:
brute-force algorithm 算法的核心思想如下, 算法的时间复杂度为O(n^3).
关于3 sum 的分析:
进一步的, 我们有1 -sum 的相关brute-force 算法如下:
1-sum: 算法复杂度: O(N)
2-sum: 算法复杂度: O(n^2)
推广到k sum problem, 算法的时间复杂度为O(n^k)。
(2)不难看出, brute -force的算法的时间复杂度太大。 我们需要更加efficient的algorithms。
我们有: 3 sum 的另外几个解决办法:利用好二分搜索。
思路是: a + b + c = 0, 所以有 c = -(a + b ), 所以我们可以嵌入二叉搜索的办法。 Binary search!! 当然, 第一步就是sorting!
更高级的, 我们可以使用一个hash Table, 可以在O(1)的时间内查找hash table 中是否有元素 (A[i] + A[j]), 是的我们算法的时间复杂度降低到O(N^2)。
注意思想参见下面的代码:
另外关于3 -sum: