leetcode 16. 最接近的三數之和

給定一個包括 n 個整數的數組 nums 和 一個目標值 target。找出 nums 中的三個整數,使得它們的和與 target 最接近。返回這三個數的和。假定每組輸入只存在唯一答案。

例如,給定數組 nums = [-1,2,1,-4], 和 target = 1.

與 target 最接近的三個數的和爲 2. (-1 + 2 + 1 = 2).

本題與上一題leetcode 15.三數之和https://blog.csdn.net/u011750466/article/details/80218002解法思路相同,程序稍有改動。

1.去重版

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int sum=0,close=INT_MAX,m=0,result=0;//轉換爲兩數之和的問題
        for(int i=0;i<nums.size()-2;++i){
            m=target-nums[i];
            int l=i+1,r=nums.size()-1;//雙指針法
            while(l<r){
                sum=nums[l]+nums[r];
                if(sum<m){
                    if(m-sum<close){
                        close=m-sum;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    ++l;
                    while(l<r&&nums[l]==nums[l-1]) ++l;//去重
                }
                else if(sum>m){
                    if(sum-m<close){
                        close=sum-m;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    --r;
                    while(l<r&&nums[r]==nums[r+1]) --r;//去重
                }
                else return target;//如果相等直接返回
            }
            while(i<nums.size()-2&&nums[i]==nums[i+1]) ++i;//去重
        }
        return result;
    }
};

2.不去重版

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int sum=0,close=INT_MAX,m=0,result=0;
        for(int i=0;i<nums.size()-2;++i){
            m=target-nums[i];
            int l=i+1,r=nums.size()-1;
            while(l<r){
                sum=nums[l]+nums[r];
                if(sum<m){
                    if(m-sum<close){
                        close=m-sum;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    ++l;
                }
                else if(sum>m){
                    if(sum-m<close){
                        close=sum-m;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    --r;
                }
                else{
                    return target;
                }
            }
        }
        return result;
    }
};

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章