Question

1865. 找出和为指定值的下标对

English Version

题目描述

给你两个整数数组 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 次

解法

方法一：哈希表

我们可以用哈希表 cnt 统计数组 nums2 中每个数字出现的次数。

对于 add 操作，我们需要更新哈希表中 nums2[index] 的值，即 cnt[nums2[index]] -= 1，然后更新 nums2[index] += val，最后更新哈希表中 nums2[index] 的值，即 cnt[nums2[index]] += 1。

对于 count 操作，我们遍历数组 nums1，对于每个数字 v，我们需要统计满足 tot - v 的数字出现的次数，即 cnt[tot - v]，然后将其累加到答案中。

时间复杂度：对于 add 操作，时间复杂度为 $O(1)$，对于 count 操作，时间复杂度为 $O(n)$，其中 $n$ 为数组 nums1 的长度。空间复杂度 $O(m)$，其中 $m$ 为数组 nums2 的长度。

class FindSumPairs:
  def __init__(self, nums1: List[int], nums2: List[int]):
    self.nums1 = nums1
    self.nums2 = nums2
    self.cnt = Counter(nums2)

  def add(self, index: int, val: int) -> None:
    old = self.nums2[index]
    self.cnt[old] -= 1
    self.cnt[old + val] += 1
    self.nums2[index] += val

  def count(self, tot: int) -> int:
    return sum(self.cnt[tot - v] for v in self.nums1)


# Your FindSumPairs object will be instantiated and called as such:
# obj = FindSumPairs(nums1, nums2)
# obj.add(index,val)
# param_2 = obj.count(tot)

class FindSumPairs {
  private int[] nums1;
  private int[] nums2;
  private Map<Integer, Integer> cnt = new HashMap<>();

  public FindSumPairs(int[] nums1, int[] nums2) {
    this.nums1 = nums1;
    this.nums2 = nums2;
    for (int v : nums2) {
      cnt.put(v, cnt.getOrDefault(v, 0) + 1);
    }
  }

  public void add(int index, int val) {
    int old = nums2[index];
    cnt.put(old, cnt.get(old) - 1);
    cnt.put(old + val, cnt.getOrDefault(old + val, 0) + 1);
    nums2[index] += val;
  }

  public int count(int tot) {
    int ans = 0;
    for (int v : nums1) {
      ans += cnt.getOrDefault(tot - v, 0);
    }
    return ans;
  }
}

/**
 * Your FindSumPairs object will be instantiated and called as such:
 * FindSumPairs obj = new FindSumPairs(nums1, nums2);
 * obj.add(index,val);
 * int param_2 = obj.count(tot);
 */

class FindSumPairs {
public:
  FindSumPairs(vector<int>& nums1, vector<int>& nums2) {
    this->nums1 = nums1;
    this->nums2 = nums2;
    for (int& v : nums2) {
      ++cnt[v];
    }
  }

  void add(int index, int val) {
    int old = nums2[index];
    --cnt[old];
    ++cnt[old + val];
    nums2[index] += val;
  }

  int count(int tot) {
    int ans = 0;
    for (int& v : nums1) {
      ans += cnt[tot - v];
    }
    return ans;
  }

private:
  vector<int> nums1;
  vector<int> nums2;
  unordered_map<int, int> cnt;
};

/**
 * Your FindSumPairs object will be instantiated and called as such:
 * FindSumPairs* obj = new FindSumPairs(nums1, nums2);
 * obj->add(index,val);
 * int param_2 = obj->count(tot);
 */

type FindSumPairs struct {
  nums1 []int
  nums2 []int
  cnt   map[int]int
}

func Constructor(nums1 []int, nums2 []int) FindSumPairs {
  cnt := map[int]int{}
  for _, v := range nums2 {
    cnt[v]++
  }
  return FindSumPairs{nums1, nums2, cnt}
}

func (this *FindSumPairs) Add(index int, val int) {
  old := this.nums2[index]
  this.cnt[old]--
  this.cnt[old+val]++
  this.nums2[index] += val
}

func (this *FindSumPairs) Count(tot int) (ans int) {
  for _, v := range this.nums1 {
    ans += this.cnt[tot-v]
  }
  return
}

/**
 * Your FindSumPairs object will be instantiated and called as such:
 * obj := Constructor(nums1, nums2);
 * obj.Add(index,val);
 * param_2 := obj.Count(tot);
 */