【Leetcode】16. 最接近的三数之和

题目描述

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数，使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

例如，给定数组 nums = [-1，2，1，-4], 和 target = 1.

与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

算法设计

用三个指针i，j，k以i=0, j=i+1, k=nums.length()-1为起点扫描，当三个数的和小于最小值时，更新sum。时间复杂度O(n^3)

C++代码

class Solution {
public:
    static bool cmp(const int a, const int b){
        return a<b;
    }
    
    int threeSumClosest(vector<int>& nums, int target) {
        int len = nums.size(), minus, sum=0;
        int num_i[len];
        if(len<3){
            return 0;
        }
        for(int i=0;i<len;i++){
            num_i[i] = nums[i];
        }
        sort(num_i, num_i+len, cmp);
        sum = num_i[0] + num_i[1] + num_i[2];
        minus = abs(target - sum);
        for(int i=0;i<len;i++){
            for(int j=i+1;j<len;j++){
                for(int k=len-1;k>=j+1;k--){
                    int tmpsum = num_i[i] + num_i[j] + num_i[k];
                    int delta = abs(target - tmpsum);
                    if(tmpsum == target){
                        sum = tmpsum;
                        return sum;
                    }
                    if(delta > minus){
                        continue;
                    }
                    else{
                        minus = delta;
                        sum = num_i[i] + num_i[j] + num_i[k];
                    }
                }
            }
        }
        return sum;
    }
};

优化算法

将数组排序，第二层循环使用双指针，一个指向i+1，一个指向末尾。计算三个数的和sum，如果target-sum = delta<minus则更新sum。
当sum<target时，将头指针向后移动；
当sum>target时，将尾指针向前移动；
当尾指针和头指针重叠时，结束。
优化后的时间复杂度为o(n^2)。

优化C++代码

class Solution {
public:
    static bool cmp(const int a, const int b){
        return a<b;
    }
    
    int threeSumClosest(vector<int>& nums, int target) {
        int len = nums.size(), minus, sum=0;
        int num_i[len];
        if(len<3){
            return 0;
        }
        for(int i=0;i<len;i++){
            num_i[i] = nums[i];
        }
        sort(num_i, num_i+len, cmp);
        sum = num_i[0] + num_i[1] + num_i[2];
        minus = abs(target - sum);
        for(int i=0;i<len;i++){
            int j = i+1, k = len-1;
            while(j < k && j < len && k < len){
                int tmpsum = num_i[i] + num_i[j] + num_i[k];
                int delta = abs(target - tmpsum);
                if(delta < minus){
                    minus = delta;
                    sum = tmpsum;
                }
                if(tmpsum > target){
                    k--;
                }
                else if(tmpsum < target){
                    j++;
                }
                else{
                    return sum;
                }
            }
        }
        return sum;
    }
};