首页 > 代码库 > LeetCode 017 4Sum
LeetCode 017 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)
【题意】
给定一个数组,从中取4个数使它们的和等于target, 找出所有的组合。
不能有重复;
且组合中各个数升序排列;
【思路】
思路和3Sum的思路完全相同,先确定第一个数,剩下的就又变成3Sum问题了【代码】
class Solution { public: vector<vector<int> > fourSum(vector<int> &num, int target) { vector<vector<int> >result; int size=num.size(); if(size<4)return result; sort(num.begin(), num.end()); //排序 for(int p1=0; p1<size-3; p1++){ if(p1!=0&&num[p1]==num[p1-1])continue; //第一个数排重 for(int p2=p1+1; p2<size-2; p2++){ if(p2!=p1+1 && num[p2]==num[p2-1])continue; //第二个数排重 int p3=p2+1; int p4=size-1; while(p3<p4){ if(p3!=p2+1 && num[p3]==num[p3-1]){p3++;continue;} //第三个数排重 int sum=num[p1]+num[p2]+num[p3]+num[p4]; if(sum==target){ vector<int> quadruplet; quadruplet.push_back(num[p1]); quadruplet.push_back(num[p2]); quadruplet.push_back(num[p3]); quadruplet.push_back(num[p4]); result.push_back(quadruplet); p3++;p4--; } else if(sum>target)p4--; else p3++; } } } return result; } };
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。