leetcode 4sum
对于ksum 问题:
首先对nums 进行排序,
- 2sum, 用两个指针指p,q向头和尾, 如果之和大于target,q 向前移,如果之和小于target ,p向后移。如果等于加入结果中。 如果p>q 结束
- 3sum,用于指针i指向一个基准,定义两个指针p,q, 分别指向i+1, end(),对于基准i, 计算nums[p]+nums[q] 与target-nums[i]作比较,和2sum 一样移动p,q. 遍历i
- 4sum, 用两个指针先指向基准i,j(j=i+1). 定义两个指针p,q, p=j+1, q=end(). 比较nums[p]+nums[q]与target-nums[i]-nums[j]的大小, 移动p,q. 如2sum一样。。 对i, j 进行遍历
注意: 重复字符导致,相同的结果:
使用while 判断p,q 移之前与移之后,有没有变化,如果没变化,继续移:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | class Solution { public : vector<vector< int >> fourSum(vector< int >& nums, int target) { sort(nums.begin(),nums.end()); vector<vector< int >> result; cout<<nums.size()<<endl; if ((nums.size())<4) return result; for ( int i=0;i<(nums.size()-3);i++) { for ( int j=i+1;j<(nums.size()-2);j++) { int res=target-nums[i]-nums[j]; int a=j+1; int b=nums.size()-1; while (a<b) { vector< int > temp; if (res==nums[a]+nums[b]) { temp.push_back(nums[i]); temp.push_back(nums[j]); temp.push_back(nums[a]); temp.push_back(nums[b]); result.push_back(temp); } if (nums[a]+nums[b]>res&& a<b) { b--; while (nums[b]==nums[b+1]&&a<b) b--; continue ; } a++; while (nums[a]==nums[a-1]&& a<b) { a++; } } while (nums[j+1]==nums[j]&&j<nums.size()-2) j++; } while (nums[i+1]==nums[i]&&i<nums.size()-3) i++; } return result; } }; |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步