这个题目想到了用双指针来做,但是自己想出来的方法在通过倒数第三个示例时发生了超时,因为用了3个循环,造成了超时。迫不得已看了解答,属实没想到可以直接对数组进行排序这个操作,涨了经验了- -。
思路:
- 对数组进行排序,返回一个有序数组。
- for循环遍历数组,left指针指向i+1位置,right指向数组最后位置,如果nums[i]大于0,且由于是有小到大排序,说明之后的和一定无法等于0,直接返回就可以。如果nums[i] + nums[left] + nums[right] > 0,则说明值过大,把right向左移动,同理如果nums[i] + nums[left] + nums[right] < 0,则说明值过小,把left向右移动。
- 判重方法我用了和解答不一样的方法,消耗了空间。新建一个set哈希表,由于已经排序了,所以假如有重复的则无法加入到set中,因此当
javanums[i] + nums[left] + nums[right] == 0
时,判断javaArrays.asList(nums[i], nums[left], nums[right])
这个组合是否存在于set,如果存在,则两个指针分别移动。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> lists = new ArrayList<>();
Arrays.sort(nums);
HashSet<List<Integer>> set = new HashSet<>();
for (int i = 0; i < nums.length; i++) {
if(nums[i] > 0)
return lists;
if(i > 0 && nums[i] == nums[i-1])
continue;
int left = i+1;
int right = nums.length - 1;
while(left < right){
if(nums[i] + nums[left] + nums[right] == 0){
List<Integer> list = Arrays.asList(nums[i], nums[left], nums[right]);
if(!set.contains(list)){
set.add(list);
lists.add(list);}
// while (left < right && nums[left] == nums[left + 1] ) left++;
// while (left < right && nums[right] == nums[right - 1] ) right--;
right--;
left++;
}
else if(nums[i] + nums[left] + nums[right] > 0 ){
right--;
}
else
left++;
}
}
return lists;
}
}