【Leetcode】15. 三数之和（3Sum）

最新推荐文章于 2022-03-29 19:08:59 发布

str_818

最新推荐文章于 2022-03-29 19:08:59 发布

阅读量152

点赞数 1

分类专栏： ----------算法日常---------- LeetCode 精选Top面试题数组文章标签： Leetcode 数组

本文链接：https://blog.csdn.net/qq_27124771/article/details/90762299

版权

----------算法日常---------- 同时被 3 个专栏收录

164 篇文章 3 订阅

订阅专栏

LeetCode

115 篇文章 3 订阅

订阅专栏

数组

52 篇文章 0 订阅

订阅专栏

Leetcode - 15 3Sum (Medium)

题目描述：找出数组中三数和为 0 的组合。

Given array nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

解题思路：注意去重。

public List<List<Integer>> threeSum(int[] nums) {
    Arrays.sort(nums);
    List<List<Integer>> res = new ArrayList<>();
    for(int i = 0; i < nums.length - 2; i++){
        if(i == 0 || (i > 0 && nums[i] != nums[i - 1])){
            int l = i + 1, r = nums.length - 1;
            while(l < r){
                if(nums[i] + nums[l] + nums[r] == 0){
                    res.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(nums[i] + nums[l] + nums[r] < 0){
                    l++;
                }else{
                    r--;
                }
            }
        }
    }
    return res;
}