【算法设计与分析作业题】第八周:16. 3Sum Closest

题目

C++ solution

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        int closestSum = nums[0] + nums[1] +nums[2];
        int diff = abs(closestSum - target);
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size() - 2; i++) { // 遍历至倒数第三个数即可
            int left = i + 1, right = nums.size()-1;
            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];
                int newDiff = abs(sum - target);
                if (diff > newDiff) {
                    diff = newDiff;
                    closestSum = sum;
                }
                if (sum < target) {
                    left++;
                } else {
                    right--;
                }
            }
        }
        return closestSum;
    }
};

简要题解

基本思路:为得到最接近给定值的三数之和,我们需要保证三数和与给定值之间的差的绝对值最小。
步骤:

  1. 用两个变量 diff 与 closestSum 分别记录当前的最小差值与相应的和;
  2. 将数组排序,可以通过策略减少遍历次数;
  3. 遍历数组:先确定一个数,然后用两个索引 left 和 right 来滑动寻找另外两个数,每确定两个数,我们求出此三数之和 sum,然后计算与给定值的差的绝对值并保存在 newDiff 中,若 newDiff < diff,则更新 diff 和 closestSum。根据 sum 与 target 的大小关系,决定索引如何滑动;
  4. 遍历结束后,变量 closestSum 中记录的值即为整个数组中最接近给定值的三数之和。

猜你喜欢

转载自blog.csdn.net/For_course/article/details/83480138