2021.9.11 CCF-稀疏向量

 

方法一：

 

#include<iostream>
#include<vector>
#include<unordered_map>
#include<algorithm>
#include<queue>

using namespace std;

int main() {
    int n;  //向量u和v的维度
    int a;  //向量u所含的非零值的个数
    int b;  //向量v所含的非零值的个数
    cin >> n >> a >> b;
    int index;  //下标
    int value;  //下标所对应的非零值
    long long sum = 0;    //内积
    unordered_map<int, int> maps;
    for (int i = 0; i < a; i++)
    {
        cin >> index >> value;
        maps[index] = value;
    }
    for (int i = 0; i < b; i++)
    {
        cin >> index >> value;
        if (maps.count(index) == 1)
        {
            sum += (maps[index] * value);
        }
    }
    cout << sum;
    return 0;
}

相比于其他的第二题来说，这道题的题意更好懂，看一遍就基本有思路了。

对于第一个向量u的index和value，我采用了哈希表来存储，这样在输入向量v的index和value时，只要判断出两个向量在同一个index上都有非零值，就可以确定这两个非零值就是内积的一部分。（据说这道题直接暴力搜索的话会超时）

但是本题一个较坑的点在于，内积sum如果设置为int型，会导致有样例在计算过程中溢出，解决办法就是将sum设置位long long型。这点实在没想到，以后遇到这种输出类型的题目，或许直接设置为long long型会更保险点。