[LeetCode]15. 3Sum

Description:

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

——————————————————————————————————————————————————————————————————————————

Solution:

看到这种题目很容易想到暴力算法（三重循环），但是用脚想也知道肯定TLE。思考许久无果，看了看Discuss，发现这类题目都算作是K Sum问题（有兴趣可以自行搜索）。

类似Container with most water的思路：

1.对输入数组nums先排好序

2.从小到大每次固定一个数 i，在剩下的数组元素中设立left标记和right标记

3.若left<right，判断nums[left]+nums[right]+nums[i]与0的关系，否则返回2

4.若相等则推入vector，并令left++，right--（注意：最后还要判断nums[left]和nums[left-1]是否相同，若相同需要再令left++，以消除重复结果，right同理），返回3

5.若<0则说明nums[left]太小（而不是nums[right]太小，因为right不能向大方向移，而是两边向中间靠拢），令left++，返回3

6.若>0则说明nums[right]太大（同理），令right--，返回3

时间复杂度：O(n^2)