看到这个题目,首先我有两种思路,一种同Two Sum 一样使用哈希map的方式,以第一个元素为基准,后面再当做Two Sum来做。后来发现这种解法由于不能直接的表示我们这个数组的下标,从而难以实现对重复元素的跳过。另外哈希map的优势在于不需要进行排序,而此问题肯定难以达到O(nlgn)的复杂度,所以放弃了这种思路。
另外就是用双指针的做法。因为进行排序后的数组确定一个元素作为基准后,另外两个就可以用双指针的方式进行相向集中,从而可以实现重复元素的跳过。代码如下:以i为基准,然后i + 1和末尾元素为双指针。:当满足三个数和为0时,将三个元素放入result。此时将双指针集中一位。然后为了防止重复,之后的j以及k都进行重复即跳过操作;当三个数和大于0时,应将大的指针k左移一位;同理,当三个数和小于0时,应将小的指针j右移一位。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> result = new LinkedList<>();
for (int i = 0; i < nums.length - 2; i++) {
if (i > 0 && nums[i] == nums[i - 1]){
continue;
}
int j = i + 1;
int k = nums.length - 1;
while(j < k){
if(nums[k] == -nums[i] - nums[j]){
result.add(Arrays.asList(nums[i], nums[j], nums[k]));
j++;
k--;
while((j < k) && (nums[j] == nums[j - 1])){
j++;
}
while((j < k) && (nums[k] == nums[k + 1])){
k--;
}
}else if(nums[k] > -nums[i] - nums[j]){
k--;
}else{
j++;
}
}
}
return result;
}
}