要求:给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
思路:数组遍历
- 首先对数组进行排序,使用Arrays工具类sort方法即可,排序后固定一个数 nums[i],再使用左右指针指向 nums[i]后面的两端,数字分别为 nums[L] 和 nums[R],计算三个数的和 sum判断是否满足为0,满足则添加进结果集
- 如果 nums[i]大于 0,则三数之和必然无法等于 0,结束循环
- 如果 nums[i] == nums[i-1],则说明该数字重复,会导致结果重复,所以应该跳过
- 当 sum == 0 时,nums[L] == nums[L+1] 则会导致结果重复,应该跳过,L++
- 当 sum == 0 时,nums[R] == nums[R-1] 则会导致结果重复,应该跳过,R–
- 时间复杂度:O(n^2),n为数组长度
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList(); //集合,存储数组
int len = nums.length; //数组的长度
if (nums == null || len < 3){ //数组为空或者长度小于3,则返回
return ans;
}
Arrays.sort(nums); //自动排序 从小到大
for (int i = 0; i < len; i++) {
if (nums[i] > 0) break; //当前循环数组中第一个数如果大于0,则证明所有数都大于0,直接结束循环
if (i > 0 && nums[i] == nums[i-1]) continue;//去重,停止当前这轮的循环,避免让重复数组添加进入到ans中
int left = i + 1;//左指针,右指针
int right = len - 1; //下标
while (left < right){
int sum = nums[i] + nums[left] + nums[right]; //三数之和
if (sum == 0){
ans.add(Arrays.asList(nums[i],nums[left],nums[right])); //填入进集合ans中
while (left < right && nums[left] == nums[left+1]) left++; //去重,左指针右移一位
while (left < right && nums[right] == nums[right-1]) right--; //去重,右指针左移一位
left ++;
right --;
}else if (sum > 0){ //三数和大于0,因为是按照从小到大排的数组,所以右指针指的数太大,所以左移
right--;
}else if (sum < 0){
left++;
}
}
}
return ans;
}