4sum

来源:互联网 发布:河南省保险数据 编辑:程序博客网 时间:2024/06/08 04:28

算法思路:步骤一:判断数组中元素个数,然后对数组排序;

    步骤二:遍历数组中任意的两个数,然后设置left,right标志,如果sum<target 则left右移,否则,right左移,直到 sum == target;

    步骤三:将sum==target的四个数构成的容器,存到vector<vector<int>> result 中;

    步骤四:对容器中相同的元素进行去重:

1)sort(result.begin(),result.end()); 使用函数sort 对它进行排序

2)  auto it =unique(result.begin(),result.end()); 使用函数unique对数组进行重新排列,“消除” 相邻的重复项(此刻的消除实际只是把重复的项放到的容器末尾),unique函数返回一个迭代器,指向最后一个不重复元素之后的位置;

3)erase(it,result.end());使用erase函数删除迭代器it之后的所有元素;容器中剩余部分的元素是不重复的。

注意:和3sum相比,思路基本一样,唯一的区别就是要多加一层for循环 。


vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>>  result;
        int  n = nums.size();
        sort(nums.begin(),nums.end());
        if(n<4){
            return result;
        }
        else{
            for(int i=0;i<n-1;++i){
                for(int j=i+1;j<n;++j){
                     int left=j+1;
                         int right=n-1;
                           while(left<right){
                        int sum=nums[i]+nums[j]+nums[left]+nums[right];
                          if(sum == target){
                            vector<int> temp;
                                   temp.push_back(nums[i]);
                             temp.push_back(nums[j]);
                             temp.push_back(nums[left]);
                             temp.push_back(nums[right]);
                            result.push_back(temp);
                            left++;
                                  right--;
                        }
                        else if(sum<target){
                            left++;
                        }
                        else
                            right--;
                    }
                }
               }
         }
        sort(result.begin(),result.end());
        auto it=unique(result.begin(),result.end());
        result.erase(it,result.end());
        return result;
    }

原创粉丝点击