leetcode 每日一题 1865. 找出和为指定值的下标对

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

给你两个整数数组 nums1nums2 ,请你实现一个支持下述两类查询的数据结构:

  1. 累加 ,将一个正整数加到 nums2 中指定下标对应元素上。
  2. 计数 ,统计满足 nums1[i] + nums2[j] 等于指定值的下标对 (i, j) 数目(0 <= i < nums1.length0 <= j < nums2.length)。
    实现 FindSumPairs 类:
  • FindSumPairs(int[] nums1, int[] nums2) 使用整数数组 nums1nums2 初始化 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 <= 10^5
  • 1 <= nums1[i] <= 10^9
  • 1 <= nums2[i] <= 10^5
  • 0 <= index < nums2.length
  • 1 <= val <= 10^5
  • 1 <= tot <= 10^9
  • 最多调用 addcount 函数各 1000

思路

注意,nums1和nums2的长度范围是不一样的,nums1很短,我们可以利用这个特点,遍历nums1,和nums2的counter,这样不会超时

go 复制代码
type FindSumPairs struct {  
    nums1 []int       // nums1.length <= 1000  
    nums2 []int       // nums2.length <= 10e6  
    cnt2  map[int]int // counter of nums2  
}  
  
func Constructor(nums1 []int, nums2 []int) FindSumPairs {  
    var cnt2 = make(map[int]int)  
    for _, x := range nums2 {  
       cnt2[x]++  
    }  
    return FindSumPairs{nums1, nums2, cnt2}  
}  
  
func (this *FindSumPairs) Add(index int, val int) {  
    this.cnt2[this.nums2[index]]--  
    this.nums2[index] += val  
    this.cnt2[this.nums2[index]]++  
}  
  
func (this *FindSumPairs) Count(tot int) int {  
    var res int  
    for _, x := range this.nums1 {  
       res += this.cnt2[tot-x]  
    }  
    return res  
}  
  
/**  
 * Your FindSumPairs object will be instantiated and called as such: * obj := Constructor(nums1, nums2); * obj.Add(index,val); * param_2 := obj.Count(tot); */

也可以把nums1和nums2都转为counter

go 复制代码
type FindSumPairs struct {  
    nums1 []int       // nums1.length <= 1000  
    nums2 []int       // nums2.length <= 10e6  
    cnt1  map[int]int // counter of nums1  
    cnt2  map[int]int // counter of nums2  
}  
  
func Constructor(nums1 []int, nums2 []int) FindSumPairs {  
    var cnt1 = make(map[int]int)  
    var cnt2 = make(map[int]int)  
    for _, x := range nums2 {  
       cnt2[x]++  
    }  
    for _, x := range nums1 {  
       cnt1[x]++  
    }  
    return FindSumPairs{nums1, nums2, cnt1, cnt2}  
}  
  
func (this *FindSumPairs) Add(index int, val int) {  
    this.cnt2[this.nums2[index]]--  
    this.nums2[index] += val  
    this.cnt2[this.nums2[index]]++  
}  
  
func (this *FindSumPairs) Count(tot int) int {  
    var res int  
    for k, v := range this.cnt1 {  
       res += v * (this.cnt2[tot-k])  
    }  
    return res  
}
相关推荐
微笑尅乐4 小时前
力扣350.两个数组的交集II
java·算法·leetcode·动态规划
元亓亓亓6 小时前
LeetCode热题100--994. 腐烂的橘子--中等
算法·leetcode·职场和发展
(●—●)橘子……6 小时前
记力扣2516.每种字符至少取k个 练习理解
算法·leetcode·职场和发展
可触的未来,发芽的智生7 小时前
新奇特:神经网络烘焙坊(下),万能配方的甜蜜奥义
人工智能·python·神经网络·算法·架构
学c语言的枫子7 小时前
数据结构——基本排序算法
数据结构·算法·排序算法
jghhh018 小时前
针对大尺度L1范数优化问题的MATLAB工具箱推荐与实现
开发语言·算法·matlab
And_Ii8 小时前
LeetCode 5.最长回文字符串
数据结构·算法·leetcode
未知陨落8 小时前
LeetCode:76.数据流的中位数
算法·leetcode
Asmalin8 小时前
【代码随想录day 28】 力扣 1005. K次取反后最大化的数组和
算法·leetcode·职场和发展
m0_747266099 小时前
减治法计算数组中的零个数
数据结构·算法·leetcode