18. 四数之和
1.双指针法
-
和前面的求和题目相似,这里我们用双指针法降低算法复杂度,从O(n4)到O(n3)。4个移动指针,三重遍历,在第三重使用双指针。
-
双指针法本身并不复杂,在这里关键是我们要做好剪枝 ,才是优秀的双指针方法。
-
剪枝1:去除重复元素,四个指针,每次移动之后都要和前一个数值比较,避免重复元素参与比较;
-
剪枝2:去除绝对不可能得到结果的位置:
- 确定ind1后,如果ind1+ind1后面三个元素已经>target,那后面的结果必然也大于target,跳出循环;
- 同理,确定ind1,ind2后,如果ind1+ind2+ind2后面两个元素已经>target,跳出;
- 确定ind1后,如果ind1+末尾三个元素<target,那么ind1开头的所有结果,必然都小于target,这一个ind1就不遍历,直接前进。
- 和上同理,确定ind1,ind2后,也能进行判断,无法得到结果的ind2就跳过,直接前进。
- 确定ind1后,如果ind1+ind1后面三个元素已经>target,那后面的结果必然也大于target,跳出循环;
-
class Solution { public: vector<vector<int>> fourSum(vector<int>& nums, int target) { vector<vector<int>> result; int ind1,ind2,ind3,ind4,n=nums.size(); long long now_val; sort(nums.begin(),nums.end()); for(ind1=0;ind1<n-3;ind1++){ if(ind1>0&&nums[ind1-1]==nums[ind1]){continue;} now_val=(long long)nums[ind1]+nums[ind1+1]+nums[ind1+2]+nums[ind1+3];//使用int这里有一个用例会溢出 if(now_val>target){break;}//1.ind1+ind1后面3个数加起来已经大于 target,就再也不会有新结果了 now_val=(long long)nums[ind1]+nums[n-1]+nums[n-2]+nums[n-3]; if(now_val<target){continue;}//1.ind1+最后3个数<target,当前的ind1不符合条件 for(ind2=ind1+1;ind2<n-2;ind2++){ if(ind2>ind1+1&&nums[ind2-1]==nums[ind2]){continue;} now_val=(long long)nums[ind1]+nums[ind2]+nums[ind2+1]+nums[ind2+2]; if(now_val>target){break;}//2.ind1+ind2+ind2后面2个数加起来已经大于 target,就再也不会有新结果了 now_val=(long long)nums[ind1]+nums[ind2]+nums[n-1]+nums[n-2]; if(now_val<target){continue;}//2.ind1+ind2+最后2个数加起来< target,当前的ind2不符合条件 ind3=ind2+1; ind4=n-1; while(ind3<ind4){ now_val=nums[ind1]+nums[ind2]+nums[ind3]+nums[ind4]; if(now_val<target){ ind3++; while(ind3<ind4&&nums[ind3-1]==nums[ind3]){ind3++;} }else if(now_val>target){ ind4--; while(ind4>ind3&&nums[ind4+1]==nums[ind4]){ind4--;} }else{ vector<int> v{nums[ind1],nums[ind2],nums[ind3],nums[ind4]}; result.push_back(v); ind3++; while(ind3<ind4&&nums[ind3-1]==nums[ind3]){ind3++;} ind4--; while(ind4>ind3&&nums[ind4+1]==nums[ind4]){ind4--;} } } } } return result; } };