leetcode 15 3sum 16 3sumclosest

最新推荐文章于 2019-10-08 22:26:16 发布

原创最新推荐文章于 2019-10-08 22:26:16 发布 · 298 阅读

0 ·

CC 4.0 BY-SA版权

本文详细解析了LeetCode上的3Sum与3Sum Closest问题的算法实现，通过先排序再使用双指针技巧，有效地解决了寻找特定和的三元组问题。文章提供了完整的C++代码示例，并介绍了如何去除重复解，适用于算法学习和面试准备。

https://leetcode.com/problems/3sum/

2sum：找到一组两元组，使得和为某个数

先排序，假设从小到大排序后，设头尾两个指针，若当前指针指向元素和小于零则移动指针增大元素，反之大于零则减小元素

3sum：找所有的三元组，使得三者和为零

n^2做法先排序，固定一个点，将问题转化为2sum问题

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans; 
        sort(nums.begin(), nums.end());
        vector<int>temp(3, 0);
        for(int i = 0; i < nums.size(); i++){
            if(i && nums[i] == nums[i - 1])
                continue;//去重
            int le = i + 1, ri = nums.size() - 1;
            temp[0] = nums[i];
            while(le < ri){
                int sum = nums[i] + nums[le] + nums[ri];
                if(sum < 0){
                    le++;
                }
                else if(sum > 0){
                    ri--;
                }
                else{
                    temp[1] = nums[le];
                    temp[2] = nums[ri];
                    ans.push_back(temp);
                    while(le < ri && nums[le] == temp[1]) le++;//去重
                    while(le < ri && nums[ri] == temp[2]) ri--;//去重
                }
            }
        }
        return ans;
    }
};

3sumclosest

找一个三元组使得三数之和最接近目标值思路和3sum相似

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        int ans, dis = 0x3f3f3f3f;
        sort(nums.begin(), nums.end());
        for(int i = 0; i < nums.size(); i++){
            int le = i + 1, ri = nums.size() - 1;
            while(le < ri){
                int sum = nums[i] + nums[le] + nums[ri];
                if(abs(sum - target) < dis){
                    if(sum == target)
                        return sum;
                    ans = sum;
                    dis = abs(sum - target);
                }
                if(sum > target){
                    ri--;
                }
                else{
                    le++;
                }
            }
        }
        return ans;
    }
};