给你一个包含 n
个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = [] 输出:[]
示例 3:
输入:nums = [0] 输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
题目来源于leetcod,附上漫画更容易理解
算法参考博客:漫画算法题:两数之和与三数之和_程序员小灰的博客-CSDN博客
两数之和算法思路:以第i个数为第一个数,这时候只要找第元素是trage - arr[i]的就行了,i后面的元素用哈希表存储,查找的时间复杂度就能降低到1.
public static int[] addTwoNumbers(int[] arr, int trage){
Map<Integer, Integer> map = new HashMap<>();
int length = arr.length;
// map<元素,下标>
for(int i = 1; i < length; i++){
map.put(arr[i], i);
}
for(int i = 0; i < length; i++){
int temp = trage - arr[i];
if( map.containsValue(temp) && map.get(temp) != i){
return new int[]{i, map.get(temp)};
}
}
return null;
}
三数之和算法思路:先把数据排序。也是以第i个元素为第1个元素,则寻找i之后的元素加起来是trage - arr[i]就行了,使用双指针指向i+1和length-1下标的元素,相加大于就k--,小于就j++,直到j j == k。因为leetcode要求不能有重复元素,所以在最后的return用stream()做去重。
public static List<List<Integer>> addThreeNumbers(int[] arr, int trage){
arr = Arrays.stream(arr).sorted().toArray();
int length = arr.length;
List<List<Integer>> resList = new ArrayList<>();
for(int i = 0; i < length; i++){
int temp = trage - arr[i];
int j = i + 1;
int k = length - 1;
for(; j < k;){
if(temp > (arr[j] + arr[k])){
j ++;
continue;
}
if(temp < (arr[j] + arr[k])){
k --;
continue;
}
if(temp == (arr[j] + arr[k])){
resList.add(Arrays.asList(arr[i], arr[j], arr[k]));
j ++;
k --;
continue;
}
}
}
return resList.stream().distinct().collect(Collectors.toList());
}