leetcode算法练习【15】三数之和

所有题目源代码：Git地址

题目

### 给你一个包含 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，不动的是最小的（最大也行，中间值太麻烦，情况太多）
• 去重：
  • nums[i] == nums[i-1] 的情况为最小值重复，需要去除
  • nums[left] == nums[left-1] 为中间值重复，需要去除
  • nums[right] ==nums[right+1] 为最大值重复，需要去除
• 当最小值 nums[i] >0 时则不存在满足条件的数组，跳出循环

class Solution {
        public List<List<Integer>> threeSum(int[] nums) {
            //排序
            Arrays.sort(nums);
            List<List<Integer>> ans = new ArrayList( );
            int right, left;
            for (int i = 0; i < nums.length - 2; i++) {
                if ((i >= 1 && nums[i - 1] == nums[i])) continue;
                if (nums[i] > 0) break;
                right = nums.length - 1;
                left = i + 1;
                while (right > left) {
                    if (nums[right] + nums[i] + nums[left] > 0) {
                        right--;
                    } else if (nums[right] + nums[i] + nums[left] < 0) {
                        left++;
                    } else if (nums[right] + nums[i] + nums[left] == 0) {
                        //可以匹配
                        ans.add(Arrays.asList(nums[i], nums[left], nums[right]));
                        while (left < right && nums[left] == nums[left + 1]) left++; // 去重
                        while (left < right && nums[right] == nums[right - 1]) right--; // 去重
                        right--;
                        left++;
                    }
                }
            }
            return ans;
        }
    }

复杂度计算

• 时间复杂度：O(n^2)
• 空间复杂度：O(n)