算法15.|算法15. 3Sum
Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.
Note:
The solution set must not contain duplicate triplets.
Example:
Given array nums = [-1, 0, 1, 2, -1, -4],给一个整数数组,里面会有a、b、c三个数和为0。找到所有和为0且不相同的三个数。
A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]
NOTE: 解里面不能有相同的三个数(也就是不能有重复解);
解:
先给数组排个序,从头开始遍历,i作为第一个数,然后从剩下的数的两端找剩下两个数l,r,其和大于目标值,r--,反之l++,相等,记录对应的值,过滤掉重复值的情况。为了避免重复,还要过滤掉第一个数相同的情况。
【算法15.|算法15. 3Sum】以下为代码:
public List> threeSum(int[] nums) {
Arrays.sort(nums);
List> result = new ArrayList<>();
for (int i = 0;
i < nums.length - 2;
i++) {
if (i == 0 || nums[i] != nums[i - 1]) { // 去除第一个数相同的情况
int l = i + 1, r = nums.length - 1, sum = 0 - nums[i];
while (l < r) {
if (nums[l] + nums[r] == sum) {// 记录对应的值,要去掉重复
result.add(Arrays.asList(nums[i], nums[l], nums[r]));
while (l < r && nums[l] == nums[l + 1]) {
l++;
}
while (l < r && nums[r] == nums[r - 1]) {
r--;
}
l++;
r--;
} else if (nums[l] + nums[r] < sum) {
l++;
} else {
r--;
}
}
}
}
return result;
}
推荐阅读
- 画解算法(1.|画解算法:1. 两数之和)
- Guava|Guava RateLimiter与限流算法
- 人生是什么(2015.9.8)
- 一个选择排序算法
- SG平滑轨迹算法的原理和实现
- 《算法》-图[有向图]
- LeetCode算法题-11.|LeetCode算法题-11. 盛最多水的容器(Swift)
- 虚拟DOM-Diff算法详解
- 《数据结构与算法之美》——队列
- 算法回顾(SVD在协同过滤推荐系统中的应用)