Leetcode(18)四數之和
阿新 • • 發佈:2018-12-11
題目描述
給定一個包含 n 個整數的陣列 nums 和一個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出所有滿足條件且不重複的四元組。
注意:
答案中不可以包含重複的四元組。
示例:
給定陣列 nums = [1, 0, -1, 0, -2, 2],和 target = 0。
滿足要求的四元組集合為:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]
解體思路
呃呃呃額額,跟三數之和,兩數之和類似的吧,可以使用雙指標的方法。
class Solution { public: vector<vector<int>> fourSum(vector<int>& nums, int target) { vector<vector<int>> res; sort(nums.begin(), nums.end()); if (nums.size()<4) return {}; for (int k = 0; k < nums.size()-3; ++k) { if (k > 0 && nums[k] == nums[k - 1]) continue; int ntarget = target - nums[k]; for (int j = k+1; j < nums.size()-2; ++j) { if (j > k+1 && nums[j] == nums[j - 1]) continue; int nntarget = ntarget - nums[j]; int m = j + 1, n = nums.size() - 1; while (m < n) { if (nums[m] + nums[n] == nntarget) { res.push_back({nums[k], nums[j], nums[m], nums[n]}); while (m < n && nums[m] == nums[m + 1]) ++m; while (m < n && nums[n] == nums[n - 1]) --n; ++m; --n; } else if (nums[m] + nums[n] < nntarget) ++m; else --n; } } } return res; } };