代码随想录算法07 | LC454、383、15、18

Leetcode454. 四数相加 II

题目链接：

[454. 四数相加 II - 力扣（Leetcode）]

思路：

• 类比于使用哈希表的两数相加，将四个数两两一组，即为两数相加
• 区别在于<key,value>键值对需要两层for循环

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> res;
        for(int a:nums1)
        {
            for(int b:nums2)
            {
                res[a+b]++;//记录a+b和的个数
            }
        }
        int count=0;
        for(int c:nums3)
        {
            for(int d:nums4)
            {
                if(res.find(0-c-d)!=res.end())//只需寻找对应-c-d和的个数即可
                {
                    count+=res[0-c-d];
                }
            }
        }
        return count;
    }
};

Leetcode383. 赎金信

题目链接：

[383. 赎金信 - 力扣（Leetcode）]

思路：

• 与[242. 有效的字母异位词 - 力扣（Leetcode）]类似
• 都采用数组+字母的ASCII码之差做哈希s[i]-'a'

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int res[26]={0};
        for(int i=0;i<magazine.size();i++)
        {
            res[magazine[i]-'a']++;
        }
        for(int i=0;i<ransomNote.size();i++)
        {
            res[ransomNote[i]-'a']--;
            if(res[ransomNote[i]-'a']<0)//当res中出现负数时，说明ransomNote的字母用到了magazine没有的
            {
                return false;
            }
        }
        return true;
    }
};

Leetcode15. 三数之和

题目链接：

[15. 三数之和 - 力扣（Leetcode）]

思路：

• 哈希法很麻烦，需要考虑多次去重的细节
• 使用双指针法
  • 细节也在于去重：首先对于排好序的数组，i若大于0，则此组一定不满足；其次在i向前遍历的过程中，遇到重复的数字要跳过（ 注意：判断要从第二个数开始即检查当前数字是否与前一个重复）；在遇到符合要求的i，j，k时，也要检查当前的左右指针前后是否有重复值

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

Leetcode18. 四数之和

题目链接：

[18. 四数之和 - 力扣（Leetcode）]

思路：

• 类比于三数之和
• 区别在于双指针在后两个元素（c，d），a和b看成整体即三数之和的i
• 此题目标是target，排好序后不能简单地利用nums[a]>target排除（可能有[-4,-3];target=-7这种情况)；所以多加判断条件nums[a]>0

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> res;
        sort(nums.begin(),nums.end());
        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 a=j+1;
                int b=nums.size()-1;
                while(a<b)
                {
                    if((long)nums[i]+nums[j]+nums[a]+nums[b]>target) b--;
                    else if((long)nums[i]+nums[j]+nums[a]+nums[b]<target) a++;
                    else{
                        res.push_back(vector<int>{nums[i],nums[j],nums[a],nums[b]});
                        while(a<b&&nums[b]==nums[b-1]) b--;
                        while(a<b&&nums[a]==nums[a+1]) a++;
                        b--;
                        a++;
                    }
                    
                }
              
            }
        }
        return res;
    }
};

总结

今天前两道题很好接受，后面双指针法花费很大时间才理解，整个哈希的内容之前没有系统刷过，收获很大，感觉需要二刷，继续加油！