当先锋百科网

首页 1 2 3 4 5 6 7

Description:

Given an array S of n integers, are there elements abc 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)