代码随想录算法训练营Day7 | 454.四数相加II 383. 赎金信 15. 三数之和 18. 四数之和-CSDN博客

本文链接：https://blog.csdn.net/qq_43243723/article/details/141370234

Day7

今天的四道题目前两道非常简单，通过哈希表计数就可以解决
后两个求和比较难，二刷了还是忘记了剪枝的判断，总体框架掌握了。还要继续努力。

454. 4Sum II

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> mp;
        int ans = 0;
        for(int i : nums1)
        {
            for(int j : nums2)
            {
                ++mp[i+j];
            }
        }
        
        for(int i : nums3)
        {
            for(int j : nums4)
            {
                ans += mp[0 - i - j];
            }
        }
        return ans;
    }
};

383. Ransom Note

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        if(magazine.length() < ransomNote.length())
            return false;
        unordered_map<char, int> mp;
        for(char c : magazine)
            ++mp[c];
        
        for(char c : ransomNote)
        {
            if(--mp[c] < 0)
                return false;
        }
        return true;
    }
};

15. 3Sum

忘了双指针，还在用哈希表

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        sort(nums.begin(), nums.end());

        for(int i=0; i<nums.size(); ++i)
        {
            if(nums[i] > 0)
                return ans;
            if(i > 0 && nums[i] == nums[i-1])
                continue;
            int left = i+1;
            int right = nums.size()-1;
            while(left < right)
            {
                if(nums[right] + nums[left] + nums[i] > 0)
                {
                    right--;
                }
                else if(nums[right] + nums[left] + nums[i] < 0)
                {
                    left++;
                }
                else
                {
                    ans.push_back({nums[i], nums[left], nums[right]});
                    while(left < right && nums[right] == nums[right-1])
                        --right;
                    while(left < right && nums[left] == nums[left+1])
                        ++left;
                    ++left;
                    --right;
                }
            }
        }
        return ans;
    }
};

18. 4Sum

剪枝没做好，忘记了复数+复数可能比target小，轻易判断只要>target就剪掉了

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> ans;

        for(int i=0; i<nums.size(); ++i)
        {
            if(nums[i] > target && nums[i] >= 0)
                break;
            if(i > 0 && nums[i] == nums[i-1])
                continue;
            for(int j=i+1; j<nums.size(); ++j)
            {
                if(nums[i] + nums[j] > target && nums[i] + nums[j] >= 0)
                    break;
                if(j>i+1 && nums[j] == nums[j-1])
                    continue;
                
                int left = j+1;
                int 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
                    {
                        ans.push_back({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;
                    }
                }
            }
        }
        return ans;
    }
};