【LeetCode】015.3Sum

最新推荐文章于 2025-09-10 15:02:22 发布

李晓磊

最新推荐文章于 2025-09-10 15:02:22 发布

阅读量407

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/enjoy_lilei/article/details/45876189

LeetCode 专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一种高效算法来找出数组中所有和为零的唯一三元组，通过排序数组并采用双指针法，避免了重复结果。

题目：

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.

Note:

Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.

    For example, given array S = {-1 0 1 2 -1 -4},

    A solution set is:
    (-1, 0, 1)
    (-1, -1, 2)

解答：

首先将nums按照升序排序，第一层循环中确定第一个数字，在第二层循环里，第二个和第三个数从两边向中间扫描，计算每次的结果sum

（1）sum为0，得到一组解

（2）sum > 0 ，过大，需要将第三个数往左移动

（3）sum < 0，过小，需要将第二数往右移动

为了避免重复的结果，每次移动时跳过重复的值；

代码：

import java.util.*;
public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
		Arrays.sort(nums); // non-descending order is ensured
		List<List<Integer>> ret = new ArrayList<List<Integer>>();
		for (int i = 0; i < nums.length; i++) {
			// skip duplicate
			while (i > 0 && i < nums.length && nums[i] == nums[i - 1])
				i++;
			int j = i + 1, k = nums.length - 1;
			while (j < k) {
				int sum = nums[i] + nums[j] + nums[k];
				// find one triplet
				if (sum == 0) {
					List<Integer> triplet = new ArrayList<Integer>(3);
					triplet.add(nums[i]);
					triplet.add(nums[j]);
					triplet.add(nums[k]);
					ret.add(triplet);
					j++;
					// skip duplicate
					while (j < k && nums[j] == nums[j - 1])
						j++;
					k--;
					// skip duplicate
					while (k > j && nums[k] == nums[k + 1])
						k--;
				} else if (sum > 0) {
					k--;
					while (k > j && nums[k] == nums[k + 1])
						k--;
				} else {
					j++;
					// skip duplicate
					while (j < k && nums[j] == nums[j - 1])
						j++;
				}
			}
		}
		return ret;
	}
}