【leetcode】15.整数数组，三数之和为0

最新推荐文章于 2022-02-10 11:11:28 发布

原创最新推荐文章于 2022-02-10 11:11:28 发布 · 1.4k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

[leetcode] 专栏收录该内容

2 篇文章

订阅专栏

博客围绕LeetCode题目展开，给定含n个整数的数组，要找出所有满足a + b + c = 0且不重复的三元组。解题思路为先排序数组，确定一个数，通过移动另外两个数的指针来寻找满足条件的组合，同时跳过相邻重复元素。最后给出了输出结果。

打卡第1周

题目：

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

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

解题思路：

1.先给数组排序

2.三个数字，确定一个数，即循环数组的数。sum-c=0，这样只要让a+b=sum即可

3.如果，a+b>sum，将b往右（小）移动；a+b<sum，将a往右（大）移动

4.如果i=i+1；即相邻元素重复，跳过，不输出

代码：

package test.com;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
    public static void main(String[] args) {
        Solution solution=new Solution();
        int[] a={-1,0,1,2,-1,-4,5,6,3};
        solution.threeSum(a);
    }
    public List<List<Integer>> threeSum(int[] ints) {
        //排序
        Arrays.sort(ints);

        List<List<Integer>> out = new ArrayList<>();

        for (int i = 0; i < ints.length - 2; i++) {
            if (i == 0 || ( ints[i] != ints[i - 1])) {

                int l = i + 1, r = ints.length - 1, sum = 0 - ints[i];
                while (l < r) {
                    //sum=0-a[i]
                    if (ints[l] + ints[r] == sum) {
                        out.add(Arrays.asList(ints[l],ints[i],ints[r]));
                        //重复元素，跳过
                        while (l < r && ints[l] == ints[l + 1]) l++;
                        while (l < r && ints[r] == ints[r - 1]) r--;
                        l++;
                        r--;
                    } else if (ints[l] + ints[r] < sum) {
                        //小于，左边右移一位
                        while (l < r && ints[l] == ints[l + 1]) l++;
                        l++;
                    } else {
                        //大于，右边左移一位
                        while (l < r && ints[r] == ints[r - 1]) r--;
                        r--;
                    }
                }
            }
        }
        System.out.println(out);
        return out;
    }
}