首页 > 代码库 > [LeetCode]4Sum
[LeetCode]4Sum
Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.
Note:
- Elements in a quadruplet (a,b,c,d) must be in non-descending order. (ie, a ≤ b ≤ c ≤ d)
- The solution set must not contain duplicate quadruplets.
For example, given array S = {1 0 -1 0 -2 2}, and target = 0. A solution set is: (-1, 0, 0, 1) (-2, -1, 1, 2) (-2, 0, 0, 2)
class Solution { public: vector<vector<int> > fourSum(vector<int> &num, int target) { std::vector< std::vector<int> > ans; if (num.size() < 4) { return ans; } sort(num.begin(), num.end()); //排序后,从左到右扫描;中间两个数一个从left+1,一个从right-1开始 for (int left = 0; left < num.size() - 3;) { for (int right = num.size() - 1; right > left + 2;) { int ml = left + 1; int mr = right - 1; while (mr > ml) { int temp = num[left] + num[ml] + num[mr] + num[right]; if (temp > target) { mr--; while(num[mr] == num[mr + 1]) { mr--; } } else if (temp < target) { ml++; while(num[ml] == num[ml - 1]) { ml++; } } else { std::vector<int> temp; temp.push_back(num[left]); temp.push_back(num[ml]); temp.push_back(num[mr]); temp.push_back(num[right]); ans.push_back(temp); mr--; ml++; while(ml < mr && num[mr] == num[mr + 1]) { mr--; } while(ml < mr && num[ml] == num[ml - 1]) { ml++; } } } right --; while(left < right && num[right] == num[right + 1]) { right--; } } left++; while(num[left] == num[left - 1]) { left++; } } return ans; } };
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。