LeetCode之 3Sum

最新推荐文章于 2020-12-01 15:01:17 发布

原创最新推荐文章于 2020-12-01 15:01:17 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

23 篇文章

订阅专栏

本文提供了一个解决LeetCode经典题目“三数之和”的高效算法实现，通过先排序再双指针技巧，避免了重复解并达到了O(n^2)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

开心也是一天，不开心也是一天，我选择开心

LeetCode官网：https://leetcode.com/

我是直接用GitHub授权登录的。

问题

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.（这个翻译出来还不如不翻译，直接看下面的例子比较容易理解）

举例

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

就是找出数组中三个数相加为零的不重复集合

解法

class Solution {
   public List<List<Integer>> threeSum(int[] num) {
    Arrays.sort(num);
    List<List<Integer>> res = new LinkedList<>(); 
    for (int i = 0; i < num.length-2; i++) {
        if (i == 0 || (i > 0 && num[i] != num[i-1])) {
            int lo = i+1, hi = num.length-1, sum = 0 - num[i];
            while (lo < hi) {
                if (num[lo] + num[hi] == sum) {
                    res.add(Arrays.asList(num[i], num[lo], num[hi]));
                    while (lo < hi && num[lo] == num[lo+1]) lo++;
                    while (lo < hi && num[hi] == num[hi-1]) hi--;
                    lo++; hi--;
                } else if (num[lo] + num[hi] < sum) lo++;
                else hi--;
           }
        }
    }
    return res;
}
}