【Leetcode】15. 三数之和

题目描述

// 15. 三数之和

// 给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，
// 使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

// 注意：答案中不可以包含重复的三元组。

题解

// 双指针法

// 本题较难的点是nums存在重复数，而返回的数组不能存在重复
// 为了去重，给nums先排升序，
// for循环遍历nums，索引为i，我们遍历nums[i]，然后按照i右边的空间进行双指针查找，
// 构建左右指针left=i+1，right=len-1。
// 由于已经排序，所以三数之和nums[i] + nums[left] + nums[right]如果小于0，
// 左指针右移，如果三数之和大于0，右指针左移。
// 如果三数之和正好为0，则创建temp，存入三数。之后left看看右移一位left+1是否存在与
// 已经存入temp的nums[left]相等的数字，存在则left越过该数，相当于去重。
// right也看看左移一位right-1看看是否存在与已经存入temp的nums[right]相等的数字，
// 存在则right左移越过该数字，相当于去重。
// 直到left和右边一位数不等，直到right和左边一位数不等，
// left右移一次，right左移一次，指向与存入数字不同的数。
// 如此循环，最后返回res即可。
// 
// 执行用时：24 ms, 在所有 Java 提交中击败了67.67%的用户
// 内存消耗：42.4 MB, 在所有 Java 提交中击败55.01%的用户
import java.util.HashSet;
import java.util.ArrayList;
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {		
        List<List<Integer>> res = new ArrayList<List<Integer>>();
		Arrays.sort(nums);
		for (int i = 0; i < nums.length - 2; i++) {
			if (i == 0 || (i > 0 && nums[i] != nums[i - 1])) {
				int left = i + 1;
				int right = nums.length - 1;
				while (left < right) {
					if (nums[i] + nums[left] + nums[right] == 0) {
						List<Integer> temp = new ArrayList<>();
						temp.add(nums[i]);
						temp.add(nums[left]);
						temp.add(nums[right]);
						res.add(temp);
						while (left < right && nums[left] == nums[left + 1])
							left++;
						while (left < right && nums[right] == nums[right - 1])
							right--;
						left++;
						right--;
					}
					else if (nums[i] + nums[left] + nums[right] > 0) {
						right--;
					}
					else {
						left++;
					}
				}
			}
		}
		return res;
    }
}



// 执行用时：24 ms, 在所有 Java 提交中击败了67.40%的用户
// 内存消耗：42.5 MB, 在所有 Java 提交中击败了44.00%的用户
class Solution {
    List<List<Integer>> res = new ArrayList<>();

    public List<List<Integer>> threeSum(int[] nums) {
        int len = nums.length;
        Arrays.sort(nums);
        
        for (int i = 0; i < len - 2; i++) {
            if (i != 0 && nums[i] == nums[i - 1]) continue;
            int left = i + 1;
            int right = len - 1;
            while (left < right) {
                if (nums[i] + nums[left] + nums[right] == 0) {
                    List<Integer> combination = new ArrayList<>();
                    combination.add(nums[i]);
                    combination.add(nums[left]);
                    combination.add(nums[right]);
                    res.add(combination);

                    while (left < right && nums[left] == nums[left + 1]) {
                        left++;
                    }
                    while (left < right && nums[right] == nums[right - 1]) {
                        right--;
                    }
                    left++;
                    right--;
                }
                else if (nums[i] + nums[left] + nums[right] < 0) {
                    left++;
                }
                else {
                    right--;
                }
            }
            
        }
        return res;
    }
}