PTA刷题Advanced甲级——1002.A+B for Ploynomials——Day(1)

问题描述

计算多项式的和，每个样例有两行，每一行代表一个多项式，K是非零整数，Ni是指数，aNi是系数，K范围是1到10，NK范围是0到1000，输出一行，行尾不可有空格。

题目分析

题目很简单，由于N一定<=1000，所以我们干脆创建一个长度为1005的数组a，以输入的Ni作为索引，将输入的aNi存放入a[Ni]，如果这个位置已经有值了，我们就将新输入的值加在原有数值上即可。这么做就不用考虑正负号和零的问题了，基本上可以避免所有坑，是个很直观的例子。
然后我们将数组反向输出，如果对应位置的数值为0我们就将其跳过，注意这里可能是唯一的坑点，这个0可能是我们输入的，也可能是初始化本身就有的，但无论如何我们都不能将其输出(这也是一开始我有一个点没过的原因)。一开始我使用的是map，就没有考虑跳过0的问题，但如果我们使用数组，自然而然要将0跳过，否则输出1005个是不现实的。

代码

#include <iostream>
#include <map>
#include <iomanip>
using namespace std;
int main()
{
    int k1,k2;
    cin>>k1;
    map<int,double>exp;
    int n;
    double nk;
    for(int i = 0;i < k1;i++)
    {
        cin>>n>>nk;
        if(exp.find(n)!=exp.end())
            exp[n]+=nk;
        else
            exp[n]=nk;
    }
    cin>>k2;
    for(int i = 0;i < k2;i++)
    {
        cin>>n>>nk;
        if(exp.find(n)!=exp.end())
            exp[n]+=nk;
        else
            exp[n]=nk;
        if(exp[n]==0)
            exp.erase(n);
    }
    cout<<exp.size();
    for(auto rit = exp.rbegin();rit!=exp.rend();rit++)
        cout<<" "<<rit->first<<" "<<setiosflags(ios::fixed)<<setprecision(1)<<rit->second;
    return 0;
}

答题用时15min
Q2——finish√