leetcode 15.三数之和 16.最接近三数之和

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. 对数组进行排序
2. 从左侧开始，以一个值为定值。 右侧两个指针，一个指向首，一个指向尾。
3. 首尾与定值相加： 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).

思路：排序+双指针。

1. 进行排序
2. 从左侧开始，以一个值为定值。 右侧两个指针，一个指向首，一个指向尾。
3. 首尾与定值相加，判断与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;
    }
};