leetcode#18. 4Sum
Posted 老鼠阿尔吉侬
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了leetcode#18. 4Sum相关的知识,希望对你有一定的参考价值。
给定一个包含 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] ]
//n sum的题看来上瘾了,看来要写个n可变的 sum的dfs版本。 class Solution { public: vector<vector<int>> fourSum(vector<int>& nums, int target) { if(nums.size()<4) return {}; //wow!!! //先排序,再遍历,对于每个元素,查找剩余元素里是不是存在3个数之和满足target1 //内循环,看剩余元素里是不是存在2个数之和满足target2 //内循环,看剩余元素里是不是存在1个数之和满足target3 //这样是不是可行,先看一看target=10,[12345], 1,9 sort(nums.begin(),nums.end()); vector<vector<int>> result; for(int i=0;i!=nums.size();++i)//nums.size()-3 { if(i-1>0&&nums[i]==nums[i-1]) continue;//j > i + 1 for(int j=i+1;j!=nums.size();++j) { int m=j+1; int n=nums.size()-1; while(m<n) { auto sum=nums[i]+nums[j]+nums[m]+nums[n]; if(sum==target) result.push_back({nums[i],nums[j],nums[m],nums[n]}); if(sum<target) m++; else n--; } } } sort(result.begin(),result.end()); result.erase(unique(result.begin(),result.end()),result.end()); return result; } };
以上是关于leetcode#18. 4Sum的主要内容,如果未能解决你的问题,请参考以下文章