1002 A+B for Polynomials (25分)

原题链接
注意点：
保留1位小数
记得考虑系数相加为0的情况 容器真的很好用直接删除 还可以记录数组大小 简直完美

#include <bits/stdc++.h>
using namespace std;

struct node
{
    double coe; //系数
    int exp; //指数
    bool flag;
};
vector<node> A, B, ans;
bool cmp(node A, node B)
{
    return A.exp > B.exp;
}
int main()
{
    int an, bn;
    cin >> an;
    for(int i = 1; i < an * 2; i = i + 2)
    {
        node temp;
        cin >> temp.exp >> temp.coe;
        temp.flag = false;
        A.push_back(temp);
    }

    cin >> bn;
    for(int i = 1; i < bn * 2; i = i + 2)
    {
        node temp;
        cin >> temp.exp >> temp.coe;
        temp.flag = false;
        B.push_back(temp);
    }
    for(int i = 0; i < A.size(); i++)
    {
        for(int j = 0; j < B.size(); j++)
        {
            if( B[j].exp == A[i].exp && B[j].flag == false)
            {
                B[j].flag = true;
                A[i].coe += B[j].coe;
                if(A[i].coe == 0)
                {
                    A.erase(A.begin() + i);
                }
                break;
            }
        }
    }
    ans = A;
    for(int i = 0; i < B.size(); i++)
    {
        if( B[i].flag == false)
        {
            ans.push_back(B[i]);
        }
    }
    cout << ans.size();
    sort(ans.begin(), ans.end(), cmp);
    for(int i = 0; i < ans.size(); i++)
    {
        printf(" %d %.1lf",ans[i].exp,ans[i].coe);
    }
    return 0;
}