leetcode(15)---三数之和（双指针）

Posted 2021-08-27 叶卡捷琳堡

文章目录

• • 题目
  • 题解

题目

15.三数之和
给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 请你找出所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

示例 2：

输入：nums = []
输出：[]

示例 3：

输入：nums = [0]
输出：[]

提示：

0 <= nums.length <= 3000
-105 <= nums[i] <= 105

题解

对数组进行排序后，先确定第一个元素，再利用双指针确定第二和第三个元素

class Solution 
{
    public List<List<Integer>> threeSum(int[] nums) 
    {
        List<List<Integer>> number = new ArrayList<>();
        if(nums.length < 3){
            return number;
        }
        //对数组进行排序
        Arrays.sort(nums);
        //找出第一个数
        for(int first = 0;first < nums.length - 1;first++){
            //如果这个数之前出现过，则继续循环
            if(first > 0 && nums[first] == nums[first-1]){
                continue;
            }
            //另外两个数的和
            int target = -nums[first];
            //第三个数的索引
            int third = nums.length - 1;
            for(int second = first + 1;second < third;){
                //排除重复
                if(second > first + 1 && nums[second] == nums[second-1]){
                    second++;
                    continue;
                }
                //如果两数之和比target大
                if(nums[second] + nums[third] > target){
                    third--;
                }
                else if(nums[second] + nums[third] < target){
                    second++;
                }
                else if(nums[second] + nums[third] == target){
                    List<Integer> list = new ArrayList<Integer>();
                    list.add(nums[first]);
                    list.add(nums[second]);
                    list.add(nums[third]);
                    number.add(list);
                    second++;
                    third--;
                }
            }
        }
        return number;
    }
}