1. 程式人生 > >LeetCode 16. 3Sum Closest(給定和,求三元組)

LeetCode 16. 3Sum Closest(給定和,求三元組)

題目描述:

    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;
    }
};