Leetcode—1570. 两个稀疏向量的点积【中等】Plus

源代码•宸

已于 2024-01-30 17:30:14 修改

阅读量183

点赞数 3

分类专栏： LeetCode刷题文章标签： leetcode 算法经验分享 c++ 哈希表

于 2024-01-28 22:44:52 首次发布

本文链接：https://blog.csdn.net/qq_44631615/article/details/135902166

版权

LeetCode刷题专栏收录该内容

429 篇文章

订阅专栏

本文介绍了如何使用C++实现LeetCode题目1570中两个稀疏向量的点积计算，通过SparseVector类和unordered_map数据结构实现高效计算并给出示例代码和运行结果。

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

2024每日刷题（一零四）

Leetcode—1570. 两个稀疏向量的点积

在这里插入图片描述

实现代码

class SparseVector {
public:
    
    SparseVector(vector<int> &nums) {
        for(int i = 0; i < nums.size(); i++) {
            if(nums[i]) {
                indexNum[i] = nums[i];
            }
        }
    }
    
    // Return the dotProduct of two sparse vectors
    int dotProduct(SparseVector& vec) {
        if(indexNum.size() < vec.indexNum.size()) {
            return vec.dotProduct(*this);
        }
        int ans = 0;
        for(auto& [index, num]: vec.indexNum) {
            auto it = indexNum.find(index);
            if(it != indexNum.end()) {
                ans += it->second * num;
            }
        }
        return ans;
    }
private:
    unordered_map<int, int> indexNum;
};

// Your SparseVector object will be instantiated and called as such:
// SparseVector v1(nums1);
// SparseVector v2(nums2);
// int ans = v1.dotProduct(v2);