每日一题(三、四数之和)+java学习
第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] ]
也有点收获,就是剪枝,有时候并不需要你全部遍历完,这时要剪掉多余的遍历枝
写麻了,这类题都有通法了:先排序,然后遍历n-2次,再用双指针法;另外加上一些剪枝处理即可
class Solution2 { //四数之和,解答基本固定:先排序,然后遍历n-2次,再用双指针法;另外加上一些剪枝处理即可
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();i++){
if(nums[i]>target&&nums[i]>0) break; //剪枝
if(i>0&&nums[i]==nums[i-1]) continue; //去重
for(int k=i+1;k<nums.size();k++){
if(nums[k]+nums[i]>target&&nums[i]>0) break; //剪枝
if(k>i+1&&nums[k]==nums[k-1]) continue; //去重
int right =nums.size()-1;
int left=k+1;
while (left<right){
if((long)nums[i]+nums[k]+nums[left]+nums[right]>target) right--;
else if((long)nums[i]+nums[k]+nums[left]+nums[right]<target) left++;
else {
result.push_back(vector<int>{nums[i],nums[k],nums[left],nums[right]});
while(left<right&&nums[right]==nums[right-1]) right--; //注意越界问题,越界问题的关键在于 nums[i]的调用
while(left<right&&nums[left]==nums[left+1]) left++;
right--;
left++;
}
}
}
}
return result;
}
};
java学习
异常处理快学完了