題目: LeetCode - 16. 3Sum Closest

題目說明

給一個陣列及整數 target,求陣列內任意三數相加最接近 target 的值。

解題思路

先將陣列由小到大排序。選定最左邊的那個數,接著選擇右邊兩數,li + 1 開始,r 從最後一個數開始,若這三者相加等於 target 直接回傳,否則檢查這次的值跟 target 的差距是否小於之前選定的,之後判斷若是這次的值大於 target 則代表 r 需要往前選,否則代表 l 需要往後選。

參考解法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
public:
int threeSumClosest(vector<int>& nums, int target)
{
int n = nums.size(), res = nums[0] + nums[1] + nums[2];
sort(nums.begin(), nums.end());
for(int i{}; i < n - 2; ++i)
{
if(i && nums[i] == nums[i - 1]) continue;
int l = i + 1, r = n - 1;
while(l < r)
{
int tmp = nums[i] + nums[l] + nums[r];
if(tmp == target) return tmp;
res = abs(tmp - target) < abs(res - target) ? tmp : res;
tmp > target ? --r : ++l;
}
}
return res;
}
};