今天在LeetCode刷题,看到“三数之和”的算法题,看了难易程度为:“中等”,开始认为难度还是不大的,后面也折腾了一会,也发现我们在传统的开发中,对一些性能细节不够仔细的地方,特此记录一下。
题目如下:
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请你返回所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1] 输出:[] 解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0] 输出:[[0,0,0]] 解释:唯一可能的三元组和为 0 。
提示:
3 <= nums.length <= 3000
-105 <= nums[i] <= 105
先贴我的解题算法:
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
int len = nums.length;
List<List<Integer>> arrayList = new ArrayList<>();
Set<String> set = new HashSet<>();
for (int i = 0; i < len; i++) {
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) {
List<Integer> list = Arrays.asList(nums[i], nums[l], nums[r]);
String key = list.toString();
if (!set.contains(key)) {
arrayList.add(list);
set.add(key);
}
l++;
r--;
} else if (sum < 0) {
l++;
} else {
r--;
}
}
}
return arrayList;
}
解题思路:
第一次采用了传统的去重判断,但是在提交的时候,在数组中有几千个值的时候就提示“超出时间限制”,开始百思不得其解,不断优化了去重的思路,最终通过数组排序,上一位和下一位的判断去重判断,快速的解决了重复判断的性能问题,这个也是我们在传统的开发中不注意性能执行的地方。