题目:15. 给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:三数相加等于零等同于两个数的和等于另一个数的相反数。先将数组排序,方便遍历,然后定义三个指针,sum、left、right,sum指向第一个元素、left指向sum后面的第一个元素、right指向数组的最后一个元素,如果left和right指向的元素之和小于sum指向的元素和的相反数,left右移,如果left和right指向的元素之和大于sum指向的元素和的相反数,right左移,如果相等,则三个指针指向的元素之和为0,这时还要同时将left和right往中间移动,寻找是否有其他两个数字相加等于sum指向的元素。右移sum,重复上面的过程。
Java代码:
package code;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Demo15 {
static List<List<Integer>> threeSum(int[] nums){
List<List<Integer>> result = new ArrayList<>();
if (nums == null || nums.length<3)
return result;
List<Integer> singleList = new ArrayList<>();
int sum = 0;//三个数相加等于0就代表两个数相加等于另一个数的相反数
int left;
int right; //left和right指向的元素相加等于sum的相反数
int len = nums.length;
//为了避免重复查找,首先对数组进行排序
Arrays.sort(nums);
while (sum <= len -3 && nums[sum]<=0){//最小的那个数都大于0,加上后面的两个数,肯定不等于0
left = sum+1;
right = len -1;
while (left < right){
if (-nums[sum] == nums[left] + nums[right]){
singleList.add(nums[sum]);
singleList.add(nums[left]);
singleList.add(nums[right]);
result.add(singleList);
singleList = new ArrayList<>();
while (left < right && nums[left+1] == nums[left])//去重
left++;
while (left < right && nums[right-1] == nums[right])
right--;
left++;
right--;
}else if (-nums[sum] > nums[left] + nums[right]){
while (left < right && nums[left+1] == nums[left])
left++;
left++;
}else {
while (left < right && nums[right-1] == nums[right])
right--;
right--;
}
}
while (sum <= len -3 && nums[sum+1] == nums[sum])//这里要带上sum <= len -3 的条件 否则像{0,0,0}这样的数组会数组越界
sum++;
sum++;
}
return result;
}
public static void main(String[] args) {
int[] arr = {0,0,0};
threeSum(arr);
System.out.println(threeSum(arr).toString());
}
}