题目表述:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
AC代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if (nums == null || nums.length < 3) return res;
Arrays.sort(nums);
for (int i = 0, len = nums.length; i < len - 2 && nums[i] <= 0; ++i) {
if (i > 0 && nums[i] == nums[i - 1]) continue;
int lo = i + 1;
int hi = len - 1;
int target = -nums[i];
while (lo < hi) {
int n;
if ((n = nums[lo] + nums[hi]) > target) {
hi--;
} else if (n < target) {
lo++;
} else {
res.add(Arrays.asList(nums[i], nums[lo], nums[hi]));
while (lo < hi && nums[lo + 1] == nums[lo++]) ;
while (lo < hi && nums[hi - 1] == nums[hi--]) ;
}
}
}
return res;
}
}
细节1:即时的判断输入是否合法,当输入为null或者长度小于3时,显然不可能有合法解,尽早放回。
细节2:for语句的判断条件,缩小搜索区间。
细节3:在for的循环中,i不是小于nums.length;而是在初始化时将数组的长度赋值给len,之后在使用时不是调用nums.length,而是直接使用len。提高了效率也缩小了len的作用域。
细节4:处理重复输入,这种方法是处理数组类的重复输入比较经典的用法,在LeetCode discuss区中经常出现。
细节5:if ((n = nums[lo] + nums[hi]) > target) 这种处理方法在JDK源码中经常出现,同时n也可以下面的else if中重用。
细节6: while (lo < hi && nums[lo + 1] == nums[lo++]) ;
while (lo < hi && nums[hi - 1] == nums[hi--]) ; 这两行去重很精简,需要好好体会。
最终提交的结果击败了99.64%的用户。