PAT1002 A+B for Polynomials

PAT1002 A+B for Polynomials

给定两个多项式 A 和 B，计算 A+B 的结果。

输入格式
共两行，每行包含一个多项式的信息，格式如下：

K N1 aN1 N2 aN2 … NK aNK
其中，K 表示多项式中非零项的数量，Ni 和 aNi 分别表示其中一个非零项的指数和系数。

输出格式
按照与输入相同的格式，输出 A+B 的结果。

结果中的各项的系数均保留一位小数。

数据范围
K 为整数，1≤K≤10。
Ni 为整数，0≤NK<⋯<N2<N1≤1000。
aNi 为浮点数，−100≤aNi≤100。

输入样例：
2 1 2.4 0 3.2
2 2 1.5 1 0.5
输出样例：
3 2 1.5 1 2.9 0 3.2

Code

#include<iostream>
using namespace std;
#define N 1001
float A[N],B[N],C[N];
int main(){
    int K,index;
    float value;
    cin>>K;
    for(int i=0;i<K;i++){
        cin>>index>>value;
        A[index]=value;
    }
    cin>>K;
    for(int i=0;i<K;i++){
        cin>>index>>value;
        B[index]=value;
    }
    K=0;
    for(int i=0;i<N;i++){
        C[i]=A[i]+B[i];
        if(C[i]){
            K++;
        }
    }
    cout<<K;
    for(int i=N-1;i>=0;i--){
        if(C[i]){
            printf(" %d %.1f",i,C[i]);
        }
    }
    return 0;
}