今天写CCF2020年6月的第二题,出现运行超时,是我万万没想到的......
第二题是一个稀疏向量的内积计算,题目大致意思是:
两个向量,告诉我们他们的维数n,以及包含的不为零的value个数
并告诉我们两个向量那些value对应的(index, value)
需要我们对这两个向量进行内积:对应对相乘再相加
这一个题数据量其实是有点庞大的:输入次数最高可达10^6,计算结果最高可达10^12。
和它同一次的第一题,计算结果也很大(当时就进了这个坑),所以我刚开始还是知道要用long long来存结果的。
我最初的思路很简单:
用两个map变量存两个稀疏向量的(index,value)关系,而且两个向量非零位置的index我用数组a1,a2存储
之后,以第一个向量为基准,遍历他所有的index,同时和第二个向量对应位置相乘再相加,最后输出。
结果,超时了......(而且才给了30分)
之后,我进行了两步改进:
- 输入输出改进:将之前的cin cout都改为scanf printf
- 算法改进:我不再用a1 a2来存index,第一轮输入中,我直接用m1存好第一个向量的(index,value),第二轮输入时,来一个index我就看m1中对应位置是否非零,如果是,则相乘相加。
这里可能有一个疑惑点:m1中我只在有value的index位置存了值,如果对应位置未使用,一定是0吗?(对应改进后第二轮输入中的判断语句):
经过百度查询,得到的结果是:是的。
map会进行默认构造初始化,未使用位置初始化为0。
下面是满分代码:
#include<map>
using namespace std;
int n,a,b;
map<int,int> m;
long long sum=0;
int i,j,k;
int main() {
scanf("%d%d%d",&n,&a,&b);
int index,value;
for(k=0; k<a; k++) {
scanf("%d%d",&index,&value);
m[index]=value;
}
for(j=0;j<b;j++){
scanf("%d%d",&index,&value);
if(m[index]!=0)//在输入第二个向量时,就判断对应位置在第一个向量那是不是0
//如果不是0,则进行相乘和相加
sum+=(long long)m[index]*(long long)value;
}
printf("%lld",sum);
}