题目
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]
]
分析
题目隐含的条件,不能有重复的list,不能每个小list必须从小到大排序。
我的思路有两个
1 将n^3转化为2sum。leetcode第一题就是一道2sum,我用hashmap做的
2 使用数组左右指针。
实践来看,第二种方法更优。
这个有两个测试例没过。
稍后我会分析一下时间长的原因。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
HashSet<List<Integer>> set = new HashSet<>();
List<List<Integer>> list = new ArrayList<>();
for (int i = 0; i < nums.length; i++) {
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int k = i + 1; k < nums.length; k++) {
if (map.containsKey(nums[k])) {
List<Integer> tmpList = new ArrayList<>();
tmpList.add(nums[i]);
tmpList.add(nums[k]);
tmpList.add( 0 - nums[i] - nums[k]);
Collections.sort(tmpList);
set.add(tmpList);
} else {
map.put(0 - nums[i] - nums[k], k);
}
}
}
list.addAll(set);
return list;
}
}
这是左右指针的方法,去掉了放进set去重的部分,时间更短。
那如何保证不重复呢,关键点在于排序。一个有序的数组,相同的数就会相邻。
在这个程序中 if (i == 0 || (i > 0 && nums[i] != nums[i-1]))会去掉-sum相同的情况,而while (k1 < k2 && nums[k1] == nums[k1+1]) k1++;
while (k1 < k2 && nums[k2] == nums[k2-1]) k2–;会去掉指针移动时相同的情况。
public static List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> list = new ArrayList<>();
for (int i = 0; i < nums.length - 2; i++) {
if (i == 0 || (i > 0 && nums[i] != nums[i-1])) {
int k1 = i + 1;
int k2 = nums.length - 1;
while (k2-k1>=1) {
if (nums[i] + nums[k1] + nums[k2] == 0) {
list.add(Arrays.asList(nums[i], nums[k1], nums[k2]));
while (k1 < k2 && nums[k1] == nums[k1+1]) k1++;
while (k1 < k2 && nums[k2] == nums[k2-1]) k2--;
k1++;k2--;
} else if (nums[i] + nums[k1] + nums[k2] > 0) {
k2--;
}else {
k1++;
}
}
}
}
return list;
}