LeetCode 16. 最接近的三数之和(3Sum Closest)

题目描述

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

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

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

解题思路

考虑用双指针法解题。首先将数组从小到大排序,并令最接近的数closest初始化为前三个数的和。每遍历到一个数,计算后面两个数之和比较的目标tar=target-nums[i],然后令左指针left指向其后第一个数,右指针right指向最后一个数,然后计算左右指针指向数字之和与tar的差sum。

  • 若sum为0,则说明此时三个数的和正好为target,所以直接返回target
  • 若sum的绝对值小于closest-target的绝对值,说明此时三个数的和为最接近target的数,所以更新closest
  • 若左右指针指向的数字之和小于tar,此时令左指针向右移动一位有可能让和更接近tar;否则让右指针左移一位。这样移动直到左右指针重合

代码

 1 class Solution {
 2 public:
 3     int threeSumClosest(vector<int>& nums, int target) {
 4         sort(nums.begin(),nums.end());
 5         int closest=nums[0]+nums[1]+nums[2],l=nums.size();
 6         for(int i=0;i<l-2;i++){
 7             int left=i+1,right=l-1;
 8             int tar=target-nums[i];
 9             int sum;
10             while(left<right){
11                 sum=nums[left]+nums[right]-tar;
12                 if(sum==0)
13                     return target;
14                 if(abs(sum)<abs(closest-target))
15                     closest=target+sum;
16                 if(nums[left]+nums[right]<tar)
17                     left++;
18                 else
19                     right--;
20             }
21         }
22         return closest;
23     }
24 };

猜你喜欢

转载自www.cnblogs.com/wmx24/p/9074151.html
今日推荐