18. 四数之和
18. 四数之和
给定一个包含 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]
]
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]
思路:和三数之和求解类似,排序+双指针,只不过多了一层循环,第二层循环判重,b从 a+1 开始。
代码:
class Solution { public: vector<vector<int>> fourSum(vector<int>& nums, int target) { int a, b, c, d, len = nums.size(), sum, temp; vector<vector<int>> result; vector<int> out; sort(nums.begin(), nums.end()); for(a = 0; a<len; a++) { if(a>0&&nums[a]==nums[a-1]) continue; for(b = a+1; b<len; b++) { if(b>a+1&&nums[b]==nums[b-1]) continue; sum = target-nums[a]-nums[b]; c = b+1; d = len -1; while(c<d) { temp = nums[c]+nums[d]; if(temp<sum) { c++; } else if(temp>sum) { d--; } else { out.clear(); out.push_back(nums[a]); out.push_back(nums[b]); out.push_back(nums[c]); out.push_back(nums[d]); result.push_back(out); c++; d--; } } } } result.erase(unique(result.begin(), result.end()), result.end()); return result; } };
posted on 2020-08-15 09:18 Little-Prince 阅读(109) 评论(0) 编辑 收藏 举报