首页 > 代码库 > 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)Solutions:
class Solution { public: vector<vector<int> > fourSum(vector<int> &num, int target) { int n = num.size(); vector<vector<int>> res; if(n<4) return res; heap_sort(num); for(int i=0; i<n; i++) { if(i!=0 && num[i] == num[i-1]) continue; for(int j=i+1; j<n; j++) { if(j!=i+1 && num[j] == num[j-1]) continue; int low = j+1,high=n-1; while(low < high) { int sum = num[i] + num[j] + num[low] + num[high]; if(sum > target) high--; else if(sum < target) low++; else { vector<int> temp; temp.push_back(num[i]); temp.push_back(num[j]); temp.push_back(num[low]); temp.push_back(num[high]); res.push_back(temp); low++; high--; while(low<high && num[low] == num[low-1]) low++; while(low<high && num[high] == num[high+1]) high--; } } } } return res; } void swap(vector<int>& num, int i, int j) { int temp = num[i]; num[i] = num[j]; num[j] = temp; } void sink(vector<int>& num, int s, int end) { while(s < end) { int k = 2*s+1; if(k>end) return; if(k<end && num[k+1] > num[k]) k++; if(num[s] > num[k]) return; swap(num,s,k); s = k; } } void heap_sort(vector<int>& num) { int n = num.size()-1; int i; for(i=(n-1)/2; i>=0; i--) sink(num,i,n); i = n; while(i>0) { swap(num,0,i); i--; sink(num,0,i); } } };
LeetCode--4Sum
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。