PTA 7-2 一元多项式的乘法与加法运算(STL map)

设计函数分别求两个一元多项式的乘积与和。

输入格式:

输入分2行，每行分别先给出多项式非零项的个数，再以指数递降方式输入一个多项式非零项系数和指数（绝对值均为不超过1000的整数）。数字间以空格分隔。

输出格式:

输出分2行，分别以指数递降方式输出乘积多项式以及和多项式非零项的系数和指数。数字间以空格分隔，但结尾不能有多余空格。零多项式应输出0 0。

输入样例:

4 3 4 -5 2  6 1  -2 0
3 5 20  -7 4  3 1

输出样例:

15 24 -25 22 30 21 -10 20 -21 8 35 6 -33 5 14 4 -15 3 18 2 -6 1
5 20 -4 4 -5 2 9 1 -2 0

 本人刚拿到这一题的时候是忘记了之前用链表做的方法了，然后回去学习解析，结果发现链表做太复杂了，又因学了C++所以就不想太麻烦，然后思索了一下，觉得map还是能做的，第一次试了一下sample过了，调了一下AC。上代码：

#include<bits/stdc++.h>
using namespace std;
int n, c, e;
map<int, int, greater<int> > mp1;//储存第一个多项式
map<int, int, greater<int> > mp2;//储存第二个多项式
map<int, int, greater<int> > mp3;//储存两个多项式相乘后的多项式
map<int, int, greater<int> > mp4;//储存两个多项式相加后的多项式
int main(){
    cin >> n;
    while(n--){
        cin >> c >> e;
        mp1[e] = c;//注意这边的储存手法，是用指数当作map里的key，映射系数，一位指数相等就可以相加减
    }
    cin >> n;
    while(n--){
        cin >> c >> e;
        mp2[e] = c;
    }
    map<int, int, greater<int> >::iterator i, j, k;
    for(i = mp1.begin(); i != mp1.end(); ++i){//相乘：对应每项相乘
        for(j = mp2.begin(); j != mp2.end(); ++j){
            mp3[i->first+j->first] += i->second * j->second;/
            if(mp3[i->first+j->first] == 0) mp3.erase(i->first+j->first);//如果有系数为0的项，则删除，后面也一样
        }
    }
    for(i = mp1.begin(); i != mp1.end(); ++i){//相加
        mp4[i->first] += i->second;
        if(mp4[i->first] == 0) mp4.erase(i->first);
    }
    for(j = mp2.begin(); j != mp2.end(); ++j){//相加，两个多项式加两次
        mp4[j->first] += j->second;
        if(mp4[j->first] == 0) mp4.erase(j->first);
    }
    if(!mp3.empty()){//如果它不空，即不是零多项式
        for(k = mp3.begin(); k != mp3.end(); ++k){
            cout << k->second << ' ' << k->first;
            if(k != --mp3.end())
                cout << ' ';
        }
    }else cout << "0 0";
    puts("");
    if(!mp4.empty()){
        for(k = mp4.begin(); k != mp4.end(); ++k){
            cout << k->second << ' ' << k->first;
            if(k != --mp4.end())
                cout << ' ';
        }
    }else cout << "0 0";
    puts("");
    
    return 0;
}