「csp模式试题 202006-2」稀疏向量 C/C++ 100分

目录

• 题目
• 输入输出样例
• 题解思路
• 题解源码（c/c++）

题目 ：稀疏向量

【问题描述】

对于一个 n 维整数向量 v∈Zn，其在第 index 个维度上的取值记作 Vindex。这里我们约定 index 的取值从 1 开始，即 V=(v1，v2，…，vn)。下面介绍一种向量的稀疏表示方法。

如果 V 仅在少量维度上的取值不为 0，则称其为稀疏向量。

例如当 n=10 时，V=(0,0,0,5,0,0,-3,0,0,1) 就是一个稀疏向量。

由于稀疏向量的非零值较少，我们可以通过仅存储非零值的方式来节省空间。具体来说，每个非零值都可以用一个 （index, value）对来表示，即该向量在第 index 个维度上的取值 vindex=value≠0。在上面的例子中，V 就可以表示为[(4,5)，(7,-3)，(10,1)]。

接下来给出这种稀疏表示一般化的定义。 * 对于任意一个 n 维整数向量 v∈Zn，如果其在且仅在 a 个维度上取值不为 0，则可以唯一表示为：

[(index1,value1),(index2,value2),...,(indexa,valuea)]

其中所有的 index 均为整数且满足：

1≤index1＜index2＜...＜indexa小于等于n

valuei 表示向量 V 在对应维度上 indexi 上的非零值。

给出两个 n 维整数向量 U,V∈Zⁿ 的稀疏表示，试计算他们的内积。

U·V=∑ui*vi (i:1->n)

【输入格式】

从标准输入读入数据。

输入的第一行包含用空格分割的三个正整数 n、a 和 b，其中 n 表示向量 u、v 的维数，a 和 b 分别表示两个向量所含非零值的个数。

第二行到第 a+1 行输入向量 u 的稀疏表示。第 i+1 行（1 ≤ i ≤ a）包含用空格分割的两个整数 indexi 和 valuei，表示 uindexi = valuei ≠ 0。

第 a+2 行到第 a+b+1 行输入向量 v 的稀疏表示。第 j+a+1 行（1 ≤ j ≤ b）包含用空格分割的两个整数 indexj 和 valuej，表示 vindexj = valuej ≠ 0。

【输出格式】

输出到标准输出。

输出一个整数，表示向量 u 和 v 的内积 u·v。

输入输出样例

• 输入样例1

10 3 4
4 5
7 -3
10 1
1 10
4 20
5 30
7 40

• 输出样例1

-20

样例 1 解释

U=(0,0,0,5,0,0,-3,0,0,1)

V=(10,0,0,20,30,0,40,0,0)

U·V=5*20 +(-3)*40=-20

【子任务】

输入数据保证 0<a,b<n;

向量 U 和 V 在每一维度上取值的绝对值|ui|,|vi|≤10^6 (1≤i≤n)

题解思路

看完题目相信你们都会做，但是却拿不到满分。原因有以下两个坑点。

结果result一定是long long型数据，因为每个 ui、vi 的绝对值为 10^6，说明中间的结果可能达到 10^12，如果不使用 long long 型数据就会造成数据溢出。从而答案不正确。

只需要用数组记录第一组向量 U 的数据，在第二组数据输入的时候就可以直接判断进行处理。这样可以节省空间和时间，避免超时。

使用一个 U 向量的记录型指针pointer进行辅助处理。

题解（c/c++）

#include <iostream>
using namespace std;
struct P
{
    int index,value;
};
int main() {
    int n, a, b;
    cin >> n >> a >> b;
    struct P U[a + 1];
    for (int i = 0; i < a; i++) {
        cin >> U[i].index >> U[i].value;
    }
    long long result = 0; //取值的绝对值为10^6，乘积为10^12，必须用long long型
    int pointer = 0; //U向量的记录型指针
    int index, value;
    for (int i = 0; pointer < a && i < b; i++) {
        cin >> index >> value;
        while (pointer < a && index > U[pointer].index) pointer++;
        if (index > U[a - 1].index) break; //如果V向量输入的index大于了U向量最大的index，后面的全是0不用考虑
        if (pointer >= a) break;    //U向量对应的数据已经计算完，后面V向量的输入不用再考虑
        if (index == U[pointer].index) result += U[pointer++].value * value;//计算完后pointer指针增1
    }
    cout << result;
}