n개의 정수와 하나의 대상이 있는 배열 num이 있다고 가정합니다. 합이 목표에 가장 가깝도록 num에서 세 개의 정수를 찾아야 합니다. 세 정수의 합을 반환합니다. 우리는 각 입력이 정확히 하나의 솔루션을 가질 것이라는 가정을 할 수 있습니다. 따라서 주어진 배열이 [-1,2,1,-4]이고 대상이 1이면 삼중항은 [-1,2,1]이 됩니다. 이것은 가장 가까운 합, 즉 2를 갖습니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- 배열 nums, ans :=0, diff :=Infinity, n :=nums 크기 정렬
- 0 ~ n – 1 범위의 i에 대해
- 왼쪽 :=i + 1, 오른쪽 :=n – 1
- 왼쪽 <오른쪽
- temp :=숫자[왼쪽] + 숫자[오른쪽] + 숫자[i]
- if |대상 – 임시|
- temp =target이면 temp를 반환하고, 그렇지 않으면 temp> target이면 오른쪽으로 1 감소, 그렇지 않으면 왼쪽으로 1 증가
- 반환
예(C++)
더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −
#include <bits/stdc++.h> using namespace std; class Solution { public: int threeSumClosest(vector<int>& nums, int target) { sort(nums.begin(), nums.end()); int ans = 0; int diff = INT_MAX; int n = nums.size(); for(int i = 0; i < n; i++){ int left = i + 1; int right = n - 1; while(left < right){ int temp = nums[left] + nums[right] + nums[i]; if(abs(target - temp) < diff){ ans = temp; diff = abs(target - temp); } if(temp == target)return temp; else if(temp > target) right--; else left++; } } return ans; } }; main(){ Solution ob; vector<int> v = {-1,2,1,-4}; cout << ob.threeSumClosest(v, 1); }
입력
[-1,2,1,-4] 1
출력
2