1002 A+B for Polynomials - PAT (Advanced Level) Practice

This time, you are supposed to find A+B where A and B are two polynomials.

Input Specification:

Each input file contains one test case. Each case occupies 2 lines, and each line contains the information of a polynomial:

​​

where K is the number of nonzero terms in the polynomial, ​ and ​​ () are the exponents and coefficients, respectively. It is given that .

Output Specification:

For each test case you should output the sum of A and B in one line, with the same format as the input. Notice that there must be NO extra space at the end of each line. Please be accurate to 1 decimal place.

Sample Input:

2 1 2.4 0 3.2
2 2 1.5 1 0.5

Sample Output:

3 2 1.5 1 2.9 0 3.2

solve

模拟多项式相加,使用 map 来记录指数和系数的联系,然后相同指数的系数相加,注意当系数为0时即相加抵消,该项消除。

#include<iostream>
#include<map>
#include<algorithm>
using namespace std;
map<int, double, greater<int>> m;
 
int main() {
    int k;
    for(int i = 0; i < 2; i++) {
        cin >> k;
        while(k--) {
            int n;
            double a;
            cin >> n >> a;
            m[n] += a;
            if(m[n] == 0) 
                m.erase(n);
        }
    }
    cout << m.size();
    for_each(m.begin(),m.end(), [](const auto &pair) {
        printf(" %d %.1f", pair.first, pair.second);
    });
    cout << endl;
 
    return 0;
}