双指针-四数之和

四数之和

  • 1.题目描述
  • 2.测试代码
  • 3.测试结果
  • 4.易错点

1.题目描述

题目链接
双指针-四数之和_第1张图片
解题思路:
先排序,后边去重边移动指针。
双指针-四数之和_第2张图片

2.测试代码

class Solution {
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(i>0 && nums[i]==nums[i-1]){//对nums[i]进行去重
        	continue;
       	}
            for(int j=i+1;j<nums.size();j++){
                if(j>i+1 && nums[j]==nums[j-1]){//对nums[j]进行去重
                    continue;
                }
                int left;
                int right;
                left=j+1;
                right=nums.size()-1;
                while(left<right){
                    if((long)nums[i]+nums[j]+nums[left]+nums[right]>target){
                        right--;                       
                    }else if((long)nums[i]+nums[j]+nums[left]+nums[right]<target){
                        left++;
                    }else{
                        result.push_back(vector<int>{nums[i],nums[j],nums[left],nums[right]});
                        while(left<right&&nums[right]==nums[right-1]){//对right进行去重
                            right--;
                        }
                        while(left<right&&nums[left]==nums[left+1]){//对left进行去重
                            left++;
                        }
                        right--;
                        left++;
                    }
                }
            }
        }
        return result;
    }
};

3.测试结果

双指针-四数之和_第3张图片
为了达到更好的测试效果,也可以进行一些剪枝操作,提前跳出循环。

for(int i=0;i<nums.size();i++){
	if(nums[i]>=target && nums[i]>0){//剪枝操作
		break;
	}
......
for(int j=i+1;j<nums.size();j++){//剪枝操作
	if(nums[i]+nums[j]>=target && nums[i]+nums[j]>0){
		break;
	}
......	

4.易错点

if(i>0 && nums[i]==nums[i-1]){//对nums[i]进行去重
	continue;
}

如果写成

if(i>0 && nums[i]==nums[i+1]){//对nums[i]进行去重
	continue;
}

则会漏解比如nums为(-2,-1,0,0,1,2),target为0,会漏掉解(-2,0,0,2)。因为判断0和后面的0相同时会跳过前一个0,导致出错。

部分代码参考来自此链接
https://programmercarl.com/0018.%E5%9B%9B%E6%95%B0%E4%B9%8B%E5%92%8C.html

你可能感兴趣的:(算法题,算法,c++,c语言,数据结构)