leetcode刷题记录
题目 三数之和
题目描述
给你一个包含 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]
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
题意
字面值都不一样,就算不同的,有几个三元组全返回
2sum
双指针
去重复的两种思路:
- 收集所有二元组, 过滤
- 通过好的流程设计规避重复
L移动到下一个不同的数
找二元组的代码
// nums[0...end]这个范围上,有多少个不同二元组,相加==target,全返回
// {-1,5} K = 4
// {1, 3}
public static List<List<Integer>> twoSum(int[] nums, int end, int target) {
int L = 0;
int R = end;
List<List<Integer>> ans = new ArrayList<>();
while (L < R) {
if (nums[L] + nums[R] > target) {
R--;
} else if (nums[L] + nums[R] < target) {
L++;
} else { // nums[L] + nums[R] == target
if (L == 0 || nums[L - 1] != nums[L]) {
List<Integer> cur = new ArrayList<>();
cur.add(nums[L]);
cur.add(nums[R]);
ans.add(cur);
}
L++;
}
}
return ans;
}
你只要保证第1个数字不一样后面就是二元组的问题
避免arrayList插到开头的代价有点高, 所以从右往左生成所有的三元组, 把一个数塞在最后
代码
public static List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
int N = nums.length;
List<List<Integer>> ans = new ArrayList<>();
for (int i = N - 1; i > 1; i--) { // 三元组最后一个数,是arr[i] 之前....二元组 + arr[i]
if (i == N - 1 || nums[i] != nums[i + 1]) {
List<List<Integer>> nexts = twoSum(nums, i - 1, -nums[i]);
for (List<Integer> cur : nexts) {
cur.add(nums[i]);
ans.add(cur);
}
}
}
return ans;
}