力扣(LeetCode)16. 最接近的三数之和(C++)

news2024/10/7 10:19:13

双指针

快排使 $n u m s$ 正序。

设置三个指针 $i$ 指向 $n u m s$ 第一个数，从前往后枚举 $n u m s [i]$ ， $l$ 从 $n u m s [i + 1]$ 往后，指向第二个数， $r$ 从 $n u m s . s i z e () - 1$ 往左，指向第三个数。

维护 $a n s$ 变量，保存最接近的三数之和。设三数之和 $n u m s [i] + n u m s [l] + n u m s [r] = s u m$ ，如果 $∣ s u m - t a r g e t ∣ > ∣ a n s - t a r g e t ∣$ ，则 $a n s$ 更新成 $s u m$ 。

如果 $s u m > t a r g e t$ ，为了接近 $t a r g e t$ ， $r - -$ 。同理， $s u m < t a r g e t$ ，为了接近 $t a r g e t$ ， $l + +$ 。

代码展示

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

博主致语

理解思路很重要！
欢迎读者在评论区留言，作为日更博主，看到就会回复的。

AC

复杂度分析

时间复杂度: $O(n^2)$ ， $n$ 是 $n u m s$ 的长度，枚举 $n u m s [i]$ ，同时双指针枚举 $n u m s [l] / n u m s [r]$ 的时间复杂度 $O(n^2)$ 。
空间复杂度: $O (l o g n)$ ，快排的平均空间复杂度 $O (l o g n)$ 。