LeetCode 15. 3Sum

天才小镇刷题家

于 2020-11-18 13:27:31 发布

阅读量67

点赞数

本文链接：https://blog.csdn.net/weixin_52542509/article/details/109758972

版权

LeetCode 同时被 3 个专栏收录

24 篇文章 0 订阅

订阅专栏

Two Pointers

17 篇文章 0 订阅

订阅专栏

Array

15 篇文章 0 订阅

订阅专栏

思路：采取固定一个数，同时用双指针来查找另外两个数的方式

JAVA

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

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();                //ArrayList是 JAVA的 STL
        for (int k = 0; k < nums.length - 2; k++) {
            if (nums[k] > 0) break;
            if (k > 0 && nums[k] == nums[k - 1]) continue;          //避免重复三元组
            int i = k + 1, j = nums.length - 1;
            while (i < j) {
                int sum = nums[k] + nums[i] + nums[j];
                if (sum < 0) {
                    while (i < j && nums[i] == nums[++i]) ;         //避免重复三元组
                } else if (sum > 0) {
                    while (i < j && nums[j] == nums[--j]) ;         //避免重复三元组
                } else {
                    //asList的作用是直接生成数组
                    res.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[i], nums[j])));
                    while (i < j && nums[i] == nums[++i]) ;
                    while (i < j && nums[j] == nums[--j]) ;
                }
            }
        }
        return res;
    }
}

Python

class Solution:
    def threeSum(self, nums: [int]) -> [[int]]:
        nums.sort()
        res, k = [], 0                                      # res是列表
        for k in range(len(nums) - 2):
            if nums[k] > 0:
                break
            if k > 0 and nums[k] == nums[k - 1]:
                continue
            i, j = k + 1, len(nums) - 1
            while i < j:  # 3. double pointer
                s = nums[k] + nums[i] + nums[j]
                if s < 0:
                    i += 1
                    while i < j and nums[i] == nums[i - 1]:
                        i += 1
                elif s > 0:
                    j -= 1
                    while i < j and nums[j] == nums[j + 1]:
                        j -= 1
                else:
                    res.append([nums[k], nums[i], nums[j]])
                    i += 1
                    j -= 1
                    while i < j and nums[i] == nums[i - 1]:
                        i += 1
                    while i < j and nums[j] == nums[j + 1]:
                        j -= 1
        return res