15. 三数之和
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:
排序+双指针。
- 对数组进行排序
- 从左侧开始,以一个值为定值。 右侧两个指针,一个指向首,一个指向尾。
- 首尾与定值相加: 1. =0,记录这三个值;2. >0 右边指针左移;3. <0 左边指针右移。
需要注意的点:去除重复项。特殊判断,提前结束循环。
Code:
#include <vector>
#include <algorithm>
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> sb;
if(nums.size()<3) return sb;
sort(nums.begin(),nums.end());
for(int i =0 ; i<nums.size()-2;i++){
if(nums[i]>0) break;
if(i>0 && nums[i] == nums[i-1]) continue;
int j = nums.size()-1 ;
int k = i+1;
while(j>k){
if(nums[i]+nums[j]+nums[k]>0) j--;
else if (nums[i]+nums[j]+nums[k]<0) k++;
else {
sb.push_back({nums[i],nums[j],nums[k]});
k++;
j--;
while(j>k && nums[j]== nums[j+1]) j--;
while(j>k && nums[k]== nums[k-1]) k++;
}
}
}
return sb;
}
};
16. 最接近三数之和
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
思路:排序+双指针。
- 进行排序
- 从左侧开始,以一个值为定值。 右侧两个指针,一个指向首,一个指向尾。
- 首尾与定值相加,判断与target的距离,更近则更新结果。判断和与target的关系:sum>target, 右指针–;sum<target,左指针++;如果相等,直接返回。
Code
#include<vector>
#include<algorithm>
class Solution {
public:
int threeSumClosest(vector<int>& nums, int target) {
int sb = nums[0]+nums[1]+nums[2];
sort(nums.begin(),nums.end());
int sum=0;
for(int i =0 ; i < nums.size()-2 ; i++){
int j=i+1;
int k = nums.size()-1;
while(j<k){
while(j<(k-1) && (nums[i]+nums[j]+nums[k-1]) > target) k--; //写不写时间差不多
while(j<(k-1) && (nums[i]+nums[j+1]+nums[k]) < target) j++;
sum = nums[i]+nums[j]+nums[k];
if(abs(sum - target) < abs(sb - target)) sb = sum;
//sb = abs(sum - target) > abs(sb - target) ? sb : sum;
if (sum>target) k--;
else if(sum<target) j++;
else return target;
}
}
return sb;
}
};