LeetCode 16. 3Sum Closest(給定和,求三元組)
阿新 • • 發佈:2019-01-10
題目描述:
Given an array S of n integers, find three integers in S such that the sum is closest to a given number, target. Return the sum of the three integers. You may assume that each input would have exactly one solution.
例子:
For example, given array S = {-1 2 1 -4}, and target = 1. The sum that is closest to the target is 2. (-1 + 2 + 1 = 2).
分析:
題意:給定一個包含n個整數的陣列S和目標值target,求唯一解三元組(a, b, c),使得a+b+c和target最近。
思路:此題和LeetCode 15如出一轍,只是給出了一個target目標值,我們只需要在細節上做一些改進:①此時sum等於target,則找到唯一解,直接返回;②此時sum不等於target,則判斷他們之間的差距是否減少,若減少,則更新答案ans和對應的差距delta。其它的處理方法不變。
時間複雜度為O(nlogn)+O(n²)=O(n²)。
程式碼:
#include <bits/stdc++.h> using namespace std; class Solution { private: static int cmp(const int a, const int b){ return a < b; } public: int threeSumClosest(vector<int>& nums, int target) { int n = nums.size(); // Exceptional Case: if(n <= 2){ return 0; } // sort sort(nums.begin(), nums.end(), cmp); int ans, delta = INT_MAX; int left, right; for(int i = 0; i <= n - 3;){ left = i + 1; right = n - 1; bool flag = false; while(left < right){ int sum = nums[i] + nums[left] + nums[right]; // debug // cout << "sum: " << sum << ", i: " << i << ", left: " << left << ", right: " << right << endl; if(sum == target){ ans = sum; flag = true; break; } else if(sum < target){ // check if(target - sum < delta){ delta = target - sum; ans = sum; } left++; while(left < right && nums[left] == nums[left - 1]){ left++; } } else if(sum > target){ // check if(sum - target < delta){ delta = sum - target; ans = sum; } right--; while(left < right && nums[right] == nums[right + 1]){ right--; } } } if(flag){ break; } i++; while(i <= n - 3 && nums[i] == nums[i - 1]){ i++; } } return ans; } };