代码随想录算法训练营第7天 | 454.四数相加Ⅱ、383.赎金信、15.三数之和、18.四数之和

代码随想录算法训练营第7天 | 454.四数相加Ⅱ、383.赎金信、15.三数之和、18.四数之和

---

454.四数相加Ⅱ

题目链接：454.四数相加Ⅱ

解题思路

对四个数组进行两两拆分，先统计 num1 和 num2 的各个元素之和和它们出现的次数，再遍历 num3 和 num4，就可以将本题转化为两数之和。

Step1：定义 unordered_map，key 用来存放前两个数组元素和，value 存放元素和出现的次数。
Step2：遍历前两个数组，将和出现的次数存放到 map 中。
Step3：遍历后两个数组，将满足条件的情况记录到 count 中 。

代码实现

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3,
                     vector<int>& nums4) {
        unordered_map<int, int> map;
        for (int a : nums1) {
            for (int b : nums2) {
                map[a + b]++;
            }
        }
        int count = 0;
        for (int c : nums3) {
            for (int d : nums4) {
                if (map.find(0 - (c + d)) != map.end()) {
                    count += map[0 - (c + d)];
                }
            }
        }
        return count;
    }
};

题目总结

本题不需要考虑有重复的四个元素相加等于0的情况，只需要找到四个数等于0即可。

383.赎金信

题目链接：383.赎金信

解题思路

本题和之前做过的有效的字母异位词很相似，字母异位词是要求两个字符串能否相互组成，而本题要求其中字符串中的字符能够组成另一个字符串，不要求相互组成。主要思路是先遍历 magazine，记录其中每个字母出现的次数，接着遍历 ransomNote，每遇到一个字母，在哈希中做减一操作，若数字被减少到0，则返回false，否则返回true。

代码实现

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

题目总结

本题也可以使用 map，但是 map 空间消耗比较大，数组会更加的简单高效。

15.三数之和

题目链接：15.三数之和

解题思路

• 哈希解法

与两数之和思路相似，确定 0-(a+b) 是否在数组中出现过，将符合条件的三元组放到 vector 中，题目中要求三元组不可重复马，所以需要再次进行去重操作。

• 双指针解法

首先对数组中元素进行排序处理。遍历数组，将一个指针 left 放在 i+1 位置，另一个 right 在末尾。进行 while 循环，条件为 left<right，若三数之和大于0，指针向左移；和小于0，指针向右移；等于0，记录值。两个指针同时移动，进行去重处理。

代码实现

双指针解法

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(); i++) {
            if (nums[i] > 0) {
                return result;
            }
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            int left = i + 1;
            int right = nums.size() - 1;
            while (left < right) {
                if (nums[i] + nums[left] + nums[right] > 0) {
                    right--;
                } else if (nums[i] + nums[left] + nums[right] < 0) {
                    left++;
                } else {
                    result.push_back(
                        vector<int>{nums[i], 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 result;
    }
};

题目总结

注意几个剪枝操作：在 for 循环遍历时，若前面已经遍历的元素就跳过；排序之后若数组第一个元素已经大于0，那么可以直接返回结果，因为无论如何也不可能组成三元组。

18.四数之和

题目链接：18.四数之和

解题思路

和三数之和思路完全一样，使用双指针法进行解题，两层 for 循环 nums[i]+nums[k] 为确定值。注意力扣测试用例中存在高位数，所以需要找到 (long)nums[i]+nums[k]+nums[left]+nums[right]==target 的情况。

代码实现

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

题目总结

三数之和和四数之和的关键都在于剪枝和去重的操作。