【算法日记】力扣454、383

力扣454.四数相加 II

题目描述

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

• 0 <= i, j, k, l < n
• nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

示例 1：

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2
解释：
两个元组如下：
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

示例 2：

输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

提示：

• n == nums1.length
• n == nums2.length
• n == nums3.length
• n == nums4.length
• 1 <= n <= 200
• -228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

思路

用哈希表的思路，前两个数和后两个数一组。前面这组数加和添加到map中，至于为什么是unordered_map，因为unordered_map底层使用哈希表实现而且可以输入pair对。为什么是两个数一组？因为需要使用嵌套for循环向unordered_map输入和数，四个数分为两组时间复杂度可以控制在O(n^2)。第二个嵌套for循环负责寻找后两个数的和差的数，若有就把map中的值加到count结果里。

源码

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> umap;
        int count = 0;
        for (int n1 : nums1) {
            for (int n2: nums2) {
                umap[n1 + n2]++;
            }
        }
        for (int n3 : nums3) {
            for (int n4 : nums4) {
                if (umap.find(0 - (n3 + n4)) != umap.end()) {
                    count += umap[0 - (n3 + n4)];
                }
            }
        }
        return count;
    }
};

力扣383.赎金信

题目描述

给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。

如果可以，返回 true ；否则返回 false 。

magazine 中的每个字符只能在 ransomNote 中使用一次。

示例 1：

输入：ransomNote = "a", magazine = "b"
输出：false

示例 2：

输入：ransomNote = "aa", magazine = "ab"
输出：false

示例 3：

输入：ransomNote = "aa", magazine = "aab"
输出：true

提示：

• 1 <= ransomNote.length, magazine.length <= 105
• ransomNote 和 magazine 由小写英文字母组成

思路

哈希表

源码

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        unordered_map<int, int> umap;
        for (char m : magazine) {
            umap[m-97]++;
        }
        for (char r : ransomNote) {
            auto iter = umap.find(r-97);
            if (iter != umap.end()) {
                int curr = --umap[r-97];
                if (curr < 0) return false;
            } else {
                return false;
            }
        }
        return true;
    }
};