LeetCode 1570. 两个稀疏向量的点积（哈希）

文章目录

- 1. 题目
- 2. 解题

1. 题目

给定两个稀疏向量，计算它们的点积（数量积）。

实现类 SparseVector：

SparseVector(nums) 以向量 nums 初始化对象。
dotProduct(vec) 计算此向量与 vec 的点积。

稀疏向量 是指绝大多数分量为 0 的向量。
你需要高效地存储这个向量，并计算两个稀疏向量的点积。

进阶：当其中只有一个向量是稀疏向量时，你该如何解决此问题？

示例 1：
输入：nums1 = [1,0,0,2,3], nums2 = [0,3,0,4,0]
输出：8
解释：v1 = SparseVector(nums1) , v2 = SparseVector(nums2)
v1.dotProduct(v2) = 1*0 + 0*3 + 0*0 + 2*4 + 3*0 = 8示例 2：
输入：nums1 = [0,1,0,0,0], nums2 = [0,0,0,0,2]
输出：0
解释：v1 = SparseVector(nums1) , v2 = SparseVector(nums2)
v1.dotProduct(v2) = 0*0 + 1*0 + 0*0 + 0*0 + 0*2 = 0示例 3：
输入：nums1 = [0,1,0,0,2,0,0], nums2 = [1,0,0,0,3,0,4]
输出：6提示：
n == nums1.length == nums2.length
1 <= n <= 10^5
0 <= nums1[i], nums2[i] <= 100

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/dot-product-of-two-sparse-vectors
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

使用哈希存储非0的元素，key 是下标，value 是值

class SparseVector {
public:unordered_map<int,int> m;int size = 0;SparseVector(vector<int> &nums) {size = nums.size();for(int i = 0; i < nums.size(); ++i){if(nums[i])m[i] = nums[i];}}// Return the dotProduct of two sparse vectorsint dotProduct(SparseVector& vec) {int ans = 0;for(auto& kv : vec.m){if(m.find(kv.first) != m.end()){ans += m[kv.first]*kv.second;}}return ans;}
};// Your SparseVector object will be instantiated and called as such:
// SparseVector v1(nums1);
// SparseVector v2(nums2);
// int ans = v1.dotProduct(v2);