三数之和

Posted 2021-05-20 zhtzyh2012

题目地址: https://leetcode-cn.com/problems/3sum/
题目说明:
   给你一个包含n个整数的数组nums,判断nums中是否存在三个元素a,b,c,使得a+b+c=0?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组.

题目事例:
   给定数组 nums = [-1, 0, 1, 2, -1, -4]，
   满足要求的三元组集合为：
   [
     [-1, 0, 1],
     [-1, -1, 2]
   ]

题目分析:
   答案中不可以包含重复的三元组.
 

采用三指针+排序的处理方式,时间复杂度为O(n^2),空间复杂度为O(1).

public List<List<Integer>> threeSum(int[] nums) {
    // 1.对参数进行校验
    if (nums == null) return null;
    // 2.对数据进行排序
    Arrays.sort(nums);
    // 3.定义返回值数据
    List<List<Integer>> r = new ArrayList<>();
    // 4.定义最外层的指针
    for (int k = 0; k < nums.length - 2; k++) {
        // 如果nums[k]大于0,直接跳出循环
        if (nums[k] > 0) break;
        // 如果k>0并且k == k-1的值,进行下一次循环
        if (k > 0 && nums[k] == nums[k-1]) continue;
        // 进行正常的三数和的计算
        int left = k + 1, right = nums.length - 1;
        while (left < right) {
            int sum = nums[k] + nums[left] + nums[right];
            if (sum < 0) {
                // 进行左边的left右面移动
                while (left < right && nums[left] == nums[++left]);
            } else if (sum > 0) {
                // 进行右面指针左移
                while (left < right && nums[right] == nums[--right]);
            } else {
                r.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[left], nums[right])));
                while (left < right && nums[left] == nums[++left]);
                while (left < right && nums[right] == nums[--right]);
            }
        }
    }
    return r;
}

解法思路如下面图示内容: