Problem: 15. 三数之和

思路

👨‍🏫 参考

Code

⏰ 时间复杂度: $O(n^2)$
🌎 空间复杂度: $O(1)$

class Solution {
 	public List<List<Integer>> threeSum(int[] nums)
	{
		List<List<Integer>> res = new ArrayList<>();
		int len = nums.length;
		if (len < 3)
			return res;
		Arrays.sort(nums);// 升序排序

		for (int i = 0; i < len; i++)// i选一个数（三个数中最小的数）
		{
			if (nums[i] > 0)// 最小数已经 > 0,和不可能 == 0
				break;
			// 当前数和前一个数相同，去重
			if (i > 0 && nums[i] == nums[i - 1])
				continue;

			int l = i + 1;// 选取第二个数（中间的数）
			int r = len - 1;// 选取第三个数（最大的数）
			while (l < r)
			{
				int sum = nums[i] + nums[l] + nums[r];
				if (sum == 0)
				{
					res.add(Arrays.asList(nums[i], nums[l], nums[r]));
//					 去重 l 和 r 的数
//					跳过重复的值，避免重复解
//					例：[0,-2,-2,-1,1,2,2]  中的 -2 -2 2 2 就需要去重
					while (l < r && nums[l] == nums[l + 1])
						l++;
					while (l < r && nums[r] == nums[r - 1])
						r--;
//					走到这，l 是连续相同段的最后一个，r 是连续相同段的第一个
//					再跳一次，把 nums[l] nums[r] 跳过
					l++;
					r--;
				} else if (sum < 0)// 左边的值 调大一点
					l++;
				else if (sum > 0)// 右边的值 调小一点
					r--;
			}
		}
		return res;
	}
}