题目:
A和B是两个多项式,求A+B
输入格式:
每个输入文件包含一个测试用例。每个占用2行,每行包含一个多项式的信息:
K N1 aN1 N2 aN2。。。NK
其中K是多项式中非零项的个数,Ni和aNi(i=1,2,⋯,K) 分别是指数和系数。给出了1≤K≤10,0≤NK<⋯<N2<N1≤1000
输出格式:
对于每个测试用例,与输入相同的格式在一行中输出A和B的总和。请注意,在每行的末尾不能有额外的空间。请精确到小数点后1位。
输入样例:
2 1 2.4 0 3.2
2 2 1.5 1 0.5
输出样例:
3 2 1.5 1 2.9 0 3.2
AC代码:
#include<iostream>
#include<set>
using namespace std;
set<int> s;//用集合记录非零的下标
int main(){
double a[1100]={0};
int n,id;
double value;
for(int i=0;i<2;i++){
cin>>n;
for(int j=0;j<n;j++){
cin>>id>>value;
a[id]+=value;
if(a[id]!=0){
if(s.find(id)==s.end()) s.insert(id);
}
else{
if(s.find(id)!=s.end()) s.erase(id);//相加之后归零了需要从集合中剔除
}
}
}
int count=s.size();
printf("%d",count);
for(int i=1010;i>=0;i--){
if(a[i]!=0){
if(count>0) printf(" ");
printf("%d %.1lf",i,a[i]);
count--;
}
}
return 0;
}