代码训练营Day.7 | 454. 四数相加II、383. 赎金信、15. 三数之和、18. 四数之和

454. 四数相加II

1. LeetCode链接

https://leetcode.cn/problems/4sum-ii/description/

2. 题目描述

3. 想法

        的时候，时间最短。所以两个两个算。

        同时考察for循环的简易用法和unordered_map简易用法。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> mm;
        for (int i = 0; i < nums1.size(); i++) {
            for (int j = 0; j < nums2.size(); j++) {
                auto f = mm.find(nums1[i] + nums2[j]);
                if (f != mm.end()) {
                    f->second++;
                } else {
                    mm.insert(pair<int, int>(nums1[i] + nums2[j], 1));
                }
            }
        }
        // unordered_map<int, int> m4;
        int result = 0;
        for (int i = 0; i < nums3.size(); i++) {
            for (int j = 0; j < nums4.size(); j++) {
                auto f = mm.find(0 - nums3[i] - nums4[j]);
                if (f != mm.end()) {
                    result += f->second;
                }
            }
        }
        return result;

    }
};

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> umap; //key:a+b的数值，value:a+b数值出现的次数
        // 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中
        for (int a : A) {
            for (int b : B) {
                umap[a + b]++;
            }
        }
        int count = 0; // 统计a+b+c+d = 0 出现的次数
        // 在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就把map中key对应的value也就是出现次数统计出来。
        for (int c : C) {
            for (int d : D) {
                if (umap.find(0 - (c + d)) != umap.end()) {
                    count += umap[0 - (c + d)];
                }
            }
        }
        return count;
    }
};

383. 赎金信

1. LeetCode链接

https://leetcode.cn/problems/ransom-note/description/

2. 题目描述

3. 想法

        只有26个字母，用数组哈希表。

        注意可以在第二个循环中直接做判断return

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int word[26] = {0};
        for (int i = 0; i < magazine.size(); i++) {
            word[magazine[i] - 'a']++;
        }
        for (int i = 0; i < ransomNote.size(); i++) {
            word[ransomNote[i] - 'a']--;
            // if (word[i] < 0) return false;
        }
        for (int i = 0; i < 26; i++) {
            if (word[i] < 0) return false;
        }
        return true;
    }
};

15. 三数之和

1. LeetCode链接

https://leetcode.cn/problems/3sum/

2. 题目描述

3. 想法

        哈希表没试过，想想就要搞个很复杂的去重逻辑。

        先排序，有规律之后就好搞了。

        两重循环，第一重是三数中的第一个数，从最小数开始，一次右移充当这三个数中的第一个数。

        第二重循环，用left和right分别代表三个数中的第二个数和第三个数。

        图片裁剪自代码随想录。

        根据nums[i] + nums[left] + nums[right]的大小，移动left和right。

        注意：

        1. 第一个数——i，因为排序之后总会有重复数字相连，而每种数字的第一个在充当i时就会把所有跟nums[i]相关的都考虑完全。所以，每种数字的第一个充当完i后，i应该忽略考虑后续重复数字。

        2. 第二、三个数字同样遵循这个道理，每种数字的第一个充当完left、right后，应该忽略考虑后续重复数字。

        

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

18. 四数之和

1. LeetCode链接

https://leetcode.cn/problems/4sum/submissions/490585535/

2. 题目描述

3. 想法

        跟三数之和一样。

        注意 nums.size()是无符号整数。
​​​​​​​        四数相加可能会是长整型。