一元多项式的乘法与加法运算 （用stl 容器pair 来写）

PTA 数据结构习题 7-2 一元多项式的乘法与加法运算 (20 分)

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

输入格式:

输入分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

结尾无空行

 分析： 本题两点需要注意，一是加或乘后的多项式结果中指数相同的项需要合并，二是输出时系数为0不用输出，除非结果多项式的系数全为0，此时特判输出0 0； 最近用stl比较多，所以直接想到了pair 把指数和系数封装起来，写起来代码也还可以，由于被合并的项不需要输出，把系数设为0就可以了不需要真的删除。本题也可以用数组来做，用下标代表指数，数组值域表示系数，（注意指数可能为负数，需要下标+N，同步后移）。

代码如下：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair <int ,int> PI;
vector <PI> a1,a2,ans;
bool cmp(PI a,PI b)
{
    return a.second>b.second;
}
void Sortmerge( ) //排序并合并 
{
    sort(ans.begin(),ans.end(),cmp);
    for (int i=0;i<ans.size();i++) {
        while ( i+1<ans.size() && ans[i].second==ans[i+1].second ) {
            ans[i+1].first+=ans[i].first; ans[i].first=0; i++;
        }
    }
}
void Print( ) //输出
{
    int flag=0;
    for (int i=0;i<ans.size();i++) {  
        if ( ans[i].first!=0 ) { 
            if (flag==1) printf(" ");
            printf("%d %d",ans[i].first,ans[i].second);
            flag=1;
        }
    }
    if (flag==0) printf("0 0");
    puts("");
    ans.clear();
}
int main()
{
    int n1,n2;
    //输入数据 
    cin>>n1;
    for (int i=0;i<n1;i++) {
        int a,e;
        scanf("%d %d",&a,&e);
        a1.push_back({a,e});
    }
    cin>>n2;
    for (int i=0;i<n2;i++) {
        int a,e;
        scanf("%d %d",&a,&e);
        a2.push_back({a,e});
    }
    // 乘法 
    for (int i=0; i<n1; i++) {
        for (int j=0; j<n2; j++) {
            int a=a1[i].first*a2[j].first;
            int b=a1[i].second+a2[j].second;
            ans.push_back({a,b});
        }
    }
    Sortmerge();// 排序并合并 
    Print(); //输出
   
    // 加法 
    for (int i=0;i<n1;i++) ans.push_back(a1[i]); 
    for (int i=0;i<n2;i++) ans.push_back(a2[i]);
    Sortmerge();  // 排序合并 
    Print();       //输出
}