Leetcode 1570. Dot Product of Two Sparse Vectors [Python]

最新推荐文章于 2024-01-28 22:44:52 发布

Case_CaiNiao

最新推荐文章于 2024-01-28 22:44:52 发布

阅读量305

点赞数

分类专栏： Leetcode学习记录

本文链接：https://blog.csdn.net/sinat_30403031/article/details/122784624

版权

leetcode python 算法

Leetcode学习记录专栏收录该内容

268 篇文章

订阅专栏

这个博客介绍了一个SparseVector类的设计，用于处理包含大量零元素的向量。在__init__方法中，通过遍历输入列表nums1，用defaultdict存储非零元素及其索引。dotProduct方法计算两个SparseVector的内积，通过迭代一个向量的非零元素，查找另一个向量中对应索引的元素进行乘法并累加。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

系统设计题。要理解原本的nums2已然被转化好了。也就是之前vec.dic.items()就是nums2的idx和对应不为0的元素。而再来一个nums1，则需要在__init__函数里设置对应的dic来做转化。

class SparseVector:
    def __init__(self, nums: List[int]):
        self.dic = collections.defaultdict(int)
        for idx, num1 in enumerate(nums):
            if num1 != 0:
                self.dic[idx] = num1
                
    
    # Return the dotProduct of two sparse vectors
    def dotProduct(self, vec: 'SparseVector') -> int:
        res = 0
        for idx, num2 in vec.dic.items():
            res += num2*self.dic.get(idx, 0)
        return res