三数之和
题目介绍
三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
1 2
| 输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]]
|
示例 2:
示例 3:
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
题目解法
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
| package algorithm;
import java.util.ArrayList; import java.util.Arrays; import java.util.List;
public class ThreeSum {
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<>(); if (nums.length < 3) { return result; }
Arrays.sort(nums);
for (int i = 0; i < nums.length - 2; i++) { if (nums[i] > 0) { break; } if (i > 0 && nums[i] == nums[i - 1]) { continue; }
int k = nums.length - 1; int target = -nums[i]; for (int j = i + 1; j < nums.length - 1; j++) { if (nums[j] > target) { break; } if (j > i + 1 && nums[j] == nums[j - 1]) { continue; }
// 缩小范围 while (j < k && nums[k] + nums[j] > target) { k } // 缩小范围 if (j == k) { break; } if (nums[k] + nums[j] == target) { List<Integer> unit = new ArrayList<>(); unit.add(nums[i]); unit.add(nums[j]); unit.add(nums[k]); result.add(unit); } } }
return result; }
public static void main (String[]args){
int[] nums = new int[]{-1, 0, 1, 2, -1, -4}; System.out.println(threeSum(nums));
nums = new int[]{}; System.out.println(threeSum(nums));
nums = new int[]{0}; System.out.println(threeSum(nums)); }
}
|
打印:
思路:
双指针。