标签：4Sum 四数 target nums int 18 pEnd pMid tempVec

题目

Given an array nums of n integers and an integer target, are there elements a, b, c and d in nums such that a+ b + c + d = target ? Find all unique quadruplets in the array which gives the sum of target.
Note:
The solution set must not contain duplicate quadruplets.
Example:
Given array nums = [1, 0 -1, 0, -2, 2], and target = 0.
A solution set is:
[
 [-1, 0, 0, 1],
 [-2, -1, 1, 2],
 [-2, 0, 0, 2]
]

思路

本题思路很简单，有了前面3Sum的基础，这里只要将 target-d，然后就是3Sum的解题方法。

C++

 vector<vector<int>> fourSum(vector<int>& nums, int target) {
        
        vector<vector<int> > result;
        if(nums.size() < 4)
            return result;
        
        sort(nums.begin(),nums.end());
        
        int pMid = 0;
        int pEnd = 0;
        for(int i = 0;i<nums.size() - 3; i++){
            
            //转化成3Sum问题
            int subTarget=target - nums[i];
            if(i > 0 && nums[i] == nums[i-1])
                continue;
            
            for(int j = i + 1;j<nums.size()-2;j++){
                
                int subTarget2 = subTarget - nums[j];
                pMid = j + 1;
                pEnd = nums.size() - 1;
                if(j > i + 1 && nums[j] == nums[j-1])
                    continue;
                
                while(pMid < pEnd){
                    int sum1 = nums[pMid] + nums[pEnd];
                    
                    if(sum1 < subTarget2){
                        pMid ++;
                    }
                    else if(sum1 > subTarget2){
                        pEnd --;
                    }
                    else{
                        vector<int> tempVec(4,0);
                        tempVec[0] = nums[i];
                        tempVec[1] = nums[j];
                        tempVec[2] = nums[pMid];
                        tempVec[3] = nums[pEnd];
                        
                        result.push_back(tempVec);
                        
                        while(pMid < pEnd && nums[pMid] == nums[pMid+1]) //去除重复元素
                            pMid ++;
                        while(pMid < pEnd && nums[pEnd] == nums[pEnd -1])
                            pEnd --;
                        pMid ++;
                        pEnd --;
                    }
                }
            }
        }
        return result;
    }

Python

标签：4Sum,四数,target,nums,int,18,pEnd,pMid,tempVec
来源： https://www.cnblogs.com/Jessey-Ge/p/10993501.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。