15. 三数之和

双指针

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {

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

        /**
         * 双指针法
         * 先将数组排序,然后设置三个指针,遍历指针i从0开始,左指针从i + 1开始,右指针从数组尾开始
         * 每次计算三个数的和,如果和大于0右指针就左移,小于0左指针就右移,等于0则记录此时的值
         * 如果左右指针相遇则循环结束
         */
        if (nums.length < 3){
            return list;
        }

        Arrays.sort(nums);

        for (int i = 0; i < nums.length; i++) {

            /**
             * 如果第一个元素大于0,和肯定大于0,直接结束
             */
            if (nums[i] > 0){
                break;
            }

            /**
             * 如果当前元素已经出现过,就跳过
             */
            if (i > 0 && nums[i] == nums[i - 1]){
                continue;
            }

            int left = i + 1;
            int right = nums.length - 1;

            while (left < right) {

                if (nums[i] + nums[left] + nums[right] > 0){
                    right--;
                }
                else if (nums[i] + nums[left] + nums[right] < 0){
                    left++;
                }
                else {

                    /**
                     * 如果满足条件,将三个元素作为列表再添加进列表,Arrays.asList()方法直接将数转换为列表,但是不能修改
                     * 左右指针同时继续遍历,但要判断是否是重复元素,如果是就跳过
                     */
                    list.add(new ArrayList<>(Arrays.asList(nums[i], nums[left], nums[right])));
                    left++;
                    right--;

                    while (left < right && nums[left] == nums[left - 1]){
                        left++;
                    }

                    while (left < right && nums[right] == nums[right + 1]){
                        right--;
                    }
                }
            }
        }

        return list;
    }
}

/**
 * 时间复杂度 O(n^2)
 * 空间复杂度 O(1)
 */

https://leetcode-cn.com/problems/3sum/

posted @ 2021-12-07 14:33  振袖秋枫问红叶  阅读(27)  评论(0)    收藏  举报