【LeetCode】15. 三数之和


1.问题

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。

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

示例 1

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。

示例 2

输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。

示例 3

输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。

提示

  • 3 <= nums.length <= 3000
  • -105 <= nums[i] <= 105

2.解题思路

2.1 解法一

  • 先排序,假设第一个数就大于0了,则不用再遍历数组了,不可能存在三数均大于0且和为0.
  • 固定第一个数nums[i],然后第二个数nums[j]和第三个数nums[k]在数组索引(i, nums.length)范围:
    • 先对nums[i]去重
    • sum>0,则要减小元素,k--
    • sum<0,则要增大元素,j++
    • sum==0,保存结果,且对第二个数,第三个数去重,继续遍历

时间复杂度:O($n^2$),n 为数组长度.

2.2 解法二(优化)

固定第一个数后,在对后面两数通过双指针遍历时,先不计算三数之和,先判定第二个数的重复性,这样可以减少对第三个数的判重。因为:

  • 第二个数先去重了,当第二个数第一次满足三数之和为0,则下一次第二个数已经比原来的大,第三个数即时相同,也不满足三和为零的结果;
  • 当第二个数不满足三数之和为0,要么j++,要么k--,即使第三个数相同,也不会满足结果,继续遍历即可。

3.代码

class Solution {
	//解法一
    public static List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> ans = new ArrayList();
        int len = nums.length;
        if(nums == null || len < 3) {
        	return ans;
        }
        // 排序
        Arrays.sort(nums);
        for (int i = 0; i < len ; i++) {
        	// 如果当前数字大于0,则三数之和一定大于0,所以结束循环
            if(nums[i] > 0) {
            	break;
            }
            //对第一个数去重
            if(i > 0 && nums[i] == nums[i-1]) {
            	continue;
            }
            int L = i+1;
            int R = len-1;
            while(L < R){
                int sum = nums[i] + nums[L] + nums[R];
                if(sum == 0){
                    ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
                    //第二个数去重
                    while (L<R && nums[L] == nums[L+1]) L++;
                    // 第三个数去重
                    while (L<R && nums[R] == nums[R-1]) R--;
                    L++;
                    R--;
                }
                else if (sum < 0) 
                	L++;
                else if (sum > 0) 
                	R--;
            }
        }        
        return ans;
    }

	//解法二
	public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        int n = nums.length;
        List<List<Integer>> ans = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            //第一个数去重
            if (i > 0 && nums[i] == nums[i - 1]) 
                continue;
            int j = i + 1;
            int k = n - 1;
            while (j < k) {
                //第二个数 去重
                while (j > i + 1 && j < n && nums[j] == nums[j - 1]) 
                    j++;
                if (j >= k) 
                    break;
                int sum = nums[i] + nums[j] + nums[k];
                if (sum == 0) {
                    ans.add(Arrays.asList(nums[i], nums[j], nums[k]));
                    j++;
                } else if (sum > 0) {
                    k--;
                } else if (sum < 0) {
                    j++;
                }
            }
        }
        return ans;
    }
}

文章作者: Kezade
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 Kezade !
评论
  目录