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] ]需要我们找出 sum = 0 的三元组。这题的难点是如何找出不重复的三元组(a, b, c)。
做法是先将array进行排序。
我们让a 为最小的,
接着让 b 为 a 的后一个,让c 为最大的哪一个,两端进行逼近。
如图中,排序后的S 为 [ -4, -1, -1, 0, ,1, 2]。
此时 a = -4, b = -1, c = 2。 sum = -3 . 所以b不够大,b再右走。
a 遍历完一遍数组后,就找完所有符合要求的了。
复杂度:
时间复杂度:O(n^2)
空间复杂度:O(1)
代码:
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> result = new ArrayList<>();
for (int i = 0; i < nums.length; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int lo = i + 1;
int hi = nums.length - 1;
while (lo < hi) {
if ((nums[i] + nums[lo] + nums[hi]) == 0) {
result.add(Arrays.asList(nums[i], nums[lo], nums[hi]));
while (lo + 1 <= hi && nums[lo + 1] == nums[lo]) {
lo++;
}
while (hi - 1 >= lo && nums[hi - 1] == nums[hi]) {
hi--;
}
lo++;
hi--;
} else if ((nums[i] + nums[lo] + nums[hi]) > 0) {
hi--;
} else {
lo++;
}
}
}
return result;
}