牛客剑指offer：数组中相加和为0的三元组

最新推荐文章于 2021-05-19 22:43:52 发布

原创最新推荐文章于 2021-05-19 22:43:52 发布 · 302 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#双指针

剑指offer 同时被 2 个专栏收录

58 篇文章

订阅专栏

双指针

23 篇文章

订阅专栏

该博客主要介绍了如何使用三指针法解决数组中寻找相加和为0的三元组问题。首先对数组进行排序，然后通过遍历数组，设置两个额外指针，根据三数之和与0的比较来移动指针。在找到解时注意去重，避免重复元素的影响。最后，提供了具体的代码实现。

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

题目：数组中相加和为0的三元组

思路：三指针。

数组排序；
遍历数组，设当前位置为cur，用两个指针left、right分别指向cur的下一个位置和数组最后一个位置，判断这三个数之和与0的大小：
如果三数之和等于0，那么加入结果集，left右移直到不重复的数，right左移直到不重复；
如果三数之和小于0，left右移一步；
如果三数之和大于0，right左移一步。
重复上述过程。

去重：
若cur的值与cur-1的值相等，就跳过；
找到解时，移动left与right时，遇见重复的值也跳过。

代码：

import java.util.*;
public class Solution {
    public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        Arrays.sort(num);
        for (int cur = 0; cur < num.length - 2; cur ++) {
            if (cur - 1 >= 0 && num[cur] == num[cur - 1]) {
                continue;
            }
            int left = cur + 1, right = num.length - 1;
            while (left < right) {
                if (num[cur] + num[left] + num[right] == 0) {
                    ArrayList<Integer> array = new ArrayList<>();
                    array.add(num[cur]);
                    array.add(num[left]);
                    array.add(num[right]);
                    res.add(array);
                    left ++;
                    while(left < right && num[left] == num[left - 1]) {
                        left ++;
                    }
                    right --;
                    while (left < right && num[right] == num[right + 1]) {
                        right --;
                    }
                }
                else if (num[cur] + num[left] + num[right] > 0) {
                    right --;
                }
                else {
                    left ++;
                }
                
            }
        }
        return res;
    }
}