Day06: 454.四数相加II，383. 赎金信，15. 三数之和，18. 四数之和

目录

454.四数相加II

383. 赎金信

哈希法

15. 三数之和

双指针法

18. 四数之和

双指针法 

454.四数相加II

Leetcode-454 

本题解题步骤：

1. 首先定义 一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。
2. 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
3. 定义int变量count，用来统计 a+b+c+d = 0 出现的次数。
4. 在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
5. 最后返回统计值 count 就可以了。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> map;
        int target = 0, count = 0;
        // 循环遍历nums1,nums2
        for(int A : nums1){
            for(int B : nums2){
                map[A + B]++;
            }
        }
        // 循环遍历nums3,nums4
        for(int C : nums3){
            for(int D : nums4){
                target = 0-(C+D);
                if(map.find(target) != map.end()){
                    count+=map[target];
                }
            }
        }
        return count;
    }
};

• 时间复杂度: O(n^2)
• 空间复杂度: O(n^2)，最坏情况下A和B的值各不相同，相加产生的数字个数为 n^2

383. 赎金信

Leetcode-383 

给定一个赎金信 (ransom) 字符串和一个杂志(magazine)字符串，判断第一个字符串 ransom 能不能由第二个字符串 magazines 里面的字符构成。如果可以构成，返回 true ；否则返回 false。

(题目说明：为了不暴露赎金信字迹，要从杂志上搜索各个需要的字母，组成单词来表达意思。杂志字符串中的每个字符只能在赎金信字符串中使用一次。)

注意：

你可以假设两个字符串均只含有小写字母。

canConstruct("a", "b") -> false
canConstruct("aa", "ab") -> false
canConstruct("aa", "aab") -> true

哈希法

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int hash[26] = {0};
        // 考虑极端条件
        if(ransomNote.size()>magazine.size()){
            return false;
        }
        for(int i = 0; i < magazine.size(); i++){
            hash[magazine[i]-'a']++;
        }
        for(int i = 0; i< ransomNote.size(); i++){
            hash[ransomNote[i]-'a']--;
            if(hash[ransomNote[i]-'a']<0){
                return false;
            }
        }
        return true;
    }
};

• 时间复杂度: O(n)
• 空间复杂度: O(1)

15. 三数之和

Leetcode-15 

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

注意： 答案中不可以包含重复的三元组。

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为： [ [-1, 0, 1], [-1, -1, 2] 

双指针法

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> record;
        sort(nums.begin(), nums.end());
        // 找出a + b + c = 0
        for(int i = 0; i < nums.size(); i++){
            if(nums[i] > 0){
                return record;
            }
            // 去重第一个数
            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){
                    left++;
                }
                else if(nums[i] + nums[left] + nums[right] > 0){
                    right--;
                }
                else{
                    record.push_back(vector<int>{nums[i], nums[left], nums[right]});
                    // 去重逻辑应该放在找到一个三元组之后，对b和c去重
                    while(left<right && nums[left] == nums[left+1]){
                        left++;
                    }
                    while(left<right && nums[right] == nums[right-1]){
                        right--;
                    }
                    // 双指针收缩
                    left++;
                    right--;
                }
            }  
        }
        return record;
    }
};

• 时间复杂度: O(n^2)
• 空间复杂度: O(1)

18. 四数之和

 Leetcode-18

题意：给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：

答案中不可以包含重复的四元组。

示例： 给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。 满足要求的四元组集合为： [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]

双指针法 

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());
        for(int k = 0; k < nums.size(); k++){
            // 剪枝
            if(nums[k] >= 0 && nums[k] > target){
                break;
            }
            // 去重nums[k]
            if(k > 0 && nums[k] == nums[k - 1]){
                continue;
            }
            for(int i = k + 1; i < nums.size(); i++){
                // 二级剪枝
                if(nums[k] + nums[i] >= 0 && nums[k] + nums[i] > target){
                    break;
                }
                // 去重i
                if(i > k + 1 && nums[i] == nums[i - 1]){
                    continue;
                }
                int left = i + 1;
                int right = nums.size() - 1;
                while(left < right){
                    if((long) nums[k] + nums[i] + nums[left] + nums[right] < target){
                        left++;
                    }
                    else if((long) nums[k] + nums[i] + nums[left] + nums[right] > target){
                        right--;
                    }
                    else{
                        result.push_back(vector<int>{nums[k], nums[i], nums[left], nums[right]});
                        // 去重left & right
                        while(left < right && nums[left] == nums[left + 1]){
                            left++;
                        }
                        while(left < right && nums[right] == nums[right - 1]){
                            right--;
                        }
                        // 双指针收缩
                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
};

• 时间复杂度: O(n^3)
• 空间复杂度: O(1)

笔记参考：代码随想录