LeetCode-找出和为指定值的下标对-题解[哈希表]

「这是我参与11月更文挑战的第27天，活动详情查看：2021最后一次更文挑战」

题目要求

题目描述

给你两个整数数组 nums1 和 nums2 ，请你实现一个支持下述两类查询的数据结构：

1. 累加，将一个正整数加到 nums2 中指定下标对应元素上。
2. 计数，统计满足 nums1[i] + nums2[j] 等于指定值的下标对 (i, j) 数目（0 <= i < nums1.length 且 0 <= j < nums2.length）。

实现 FindSumPairs类：

• FindSumPairs(int[] nums1, int[] nums2) 使用整数数组 nums1 和 nums2 初始化 FindSumPairs 对象。
• void add(int index, int val) 将 val 加到 nums2[index] 上，即，执行 nums2[index] += val 。
• int count(int tot) 返回满足 nums1[i] + nums2[j] == tot 的下标对 (i, j) 数目。

示例

输入：
["FindSumPairs", "count", "add", "count", "count", "add", "add", "count"]
[[[1, 1, 2, 2, 2, 3], [1, 4, 5, 2, 5, 4]], [7], [3, 2], [8], [4], [0, 1], [1, 1], [7]]
输出：
[null, 8, null, 2, 1, null, null, 11]

解释：
FindSumPairs findSumPairs = new FindSumPairs([1, 1, 2, 2, 2, 3], [1, 4, 5, 2, 5, 4]);
findSumPairs.count(7);  // 返回 8 ; 下标对 (2,2), (3,2), (4,2), (2,4), (3,4), (4,4) 满足 2 + 5 = 7 ，下标对 (5,1), (5,5) 满足 3 + 4 = 7
findSumPairs.add(3, 2); // 此时 nums2 = [1,4,5,4,5,4]
findSumPairs.count(8);  // 返回 2 ；下标对 (5,2), (5,4) 满足 3 + 5 = 8
findSumPairs.count(4);  // 返回 1 ；下标对 (5,0) 满足 3 + 1 = 4
findSumPairs.add(0, 1); // 此时 nums2 = [2,4,5,4,5,4]
findSumPairs.add(1, 1); // 此时 nums2 = [2,5,5,4,5,4]
findSumPairs.count(7);  // 返回 11 ；下标对 (2,1), (2,2), (2,4), (3,1), (3,2), (3,4), (4,1), (4,2), (4,4) 满足 2 + 5 = 7 ，下标对 (5,3), (5,5) 满足 3 + 4 = 7
复制代码

提示

• 1 <= nums1.length <= 1000
• 1 <= nums2.length <= 105
• 1 <= nums1[i] <= 109
• 1 <= nums2[i] <= 105
• 0 <= index < nums2.length
• 1 <= val <= 105
• 1 <= tot <= 109
• 最多调用 add 和 count 函数各 1000 次

解题思路

我们可以通过题目得知，我们只需要计算和为指定值的下表对的个数，而不需要存下所有坐标对。同时，我们可以发现nums1的长度明显小于nums2的长度。因此，对于每次count计算，我们可以遍历较短的nums1数组，假设当前遍历的数为n，那么我们只需要找出nums2数组中值为tot-n的元素个数即可，而该元素的个数，我们可以在一开始就遍历一遍nums2数组，在一个map中存储每个值的个数。

代码

class FindSumPairs {
public:
    unordered_map<int, int> num_cnt1;
    unordered_map<int, int> num_cnt2;
    vector<int> num2;
    FindSumPairs(vector<int>& nums1, vector<int>& nums2) {
        for(auto& i:nums1) num_cnt1[i]++;
        for(auto& i:nums2) num_cnt2[i]++;
        num2 = nums2;
    }
    
    void add(int index, int val) {
        num_cnt2[num2[index]]--;
        num2[index] += val;
        num_cnt2[num2[index]]++;
    }
    
    int count(int tot) {
        int res = 0;
        for(auto [k,v]:num_cnt1){
            res += num_cnt2[tot-k]*v;
        }
        return res;
    }
};
复制代码