代码随想录打卡第7天

最新推荐文章于 2024-09-14 18:52:56 发布

装代码的口袋

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量2.5k

点赞数

文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/qq_43216763/article/details/128703291

版权

454. 四数相加 II

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> map1,map2;
        int res=0;
        for(int i=0;i<nums1.size();i++)
        {
            for(int j=0;j<nums1.size();j++)
            {
                map1[nums1[i]+nums2[j]]++;
                map2[nums3[i]+nums4[j]]++;
            }
        }
        for(auto x:map1)
        {
            if(map2.find(-(x.first))!=map2.end())
            {
                res+=x.second*map2[-(x.first)];
            }
        }
        return res;

    }
};

15. 三数之和

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        for(int i=0;i<nums.size()-2;i++)
        {
            int left=i+1;
            int right=nums.size()-1;
            if(i!=0&&nums[i]==nums[i-1]) continue;
            while(left<right)
            {
                if(nums[i]+nums[left]+nums[right]==0)
                {
                    res.push_back(vector<int>{nums[i],nums[left],nums[right]});
                    while(left<right&&nums[right]==nums[right-1]) right--;
                    while(left<right&&nums[left]==nums[left+1]) left++;
                    right--;
                    left++;
                }
                else if(nums[i]+nums[left]+nums[right]<0)
                left++;
                else right--;
            }
        }
        return res;

    }
};

18. 四数之和

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        if(nums.size()<4) return res;
        for(int i=0;i<nums.size()-3;i++)
        {
            for(int j=nums.size()-1;j>2;j--)
            {
                if(i!=0&&nums[i]==nums[i-1]) continue;
                if(j!=nums.size()-1&&nums[j]==nums[j+1]) continue;
                int left=i+1;
                int right=j-1;
                while(left<right)
                {
                    if((long)nums[i]+nums[j]+nums[left]+nums[right]==target)
                    {
                        res.push_back(vector<int>{nums[i],nums[j],nums[left],nums[right]});
                        while(left<right&&nums[left]==nums[left+1]) left++;
                        while(left<right&&nums[right]==nums[right-1]) right--;
                        left++;
                        right--;
                    }
                    else if((long)nums[i]+nums[j]+nums[left]+nums[right]>target) right--;
                    else left++;
                }
            }
        }
        return res;

    }
};

需要重新做的题目
15. 三数之和
18. 四数之和(与三树之和类似，注意数据类型长度）
需要学的知识点
res.push_back(vector{nums[i],nums[left],nums[right]});