最接近的三數之和
阿新 • • 發佈:2020-09-22
1.問題描述
給定一個包括n 個整數的陣列nums
和 一個目標值target
。找出nums
中的三個整數,使得它們的和與target
最接近。返回這三個數的和。假定每組輸入只存在唯一答案。
示例:
輸入:nums = [-1,2,1,-4], target = 1
輸出:2
解釋:與 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
提示:
3 <= nums.length <= 10^3
-10^3<= nums[i]<= 10^3
-10^4<= target<= 10^4
2.求解
雙指標法
本題與題目
非常類似,這裡我們用相同的思路來解決
- 先對陣列進行排序
- 遍歷陣列,固定最左邊數作為一個指標
k
,定義指標i
,j
分別從k
和陣列最右端,然後從兩端分別移動i
,j
- 令三數之和等於sum,定義整數
ans
為max
,每次更新target - sum
的值,若小於ans
則令ans = target - sum
- 當
sum > 0
時,進行j--
的操作 - 當
sum < 0
時,進行i++
的操作 - 當
sum = 0
時,跳出
- 當
程式碼如下
public int threeSumClosest(int[] nums, int target) { int len = nums.length,ans = nums[0] + nums[1] + nums[2] ; if(nums == null || len < 3) return ans; Arrays.sort(nums); for (int k = 0; k < len - 2; k++) { int i = k + 1, j = len - 1; while (i < j) { int sum = nums[k] + nums[i] + nums[j]; if (Math.abs(sum - target) < Math.abs(target - ans)) { ans = sum; } if(sum > target){ j--; } else if(sum < target) { i++; } else { return ans; } } } return ans; }
- 時間複雜度O(n²)
- 空間複雜度O(1)