当前位置:编程学习 > C/C++ >>

[PAT Advanced Level]1002. A+B for Polynomials (25)

思路:先将两个多项式存入到两个数组中,再利用归并排序的方法合并两个多项式。
注意:遇到相加后系数为0的项要舍弃,项的数目也不能+1。
改进:写到一半发现项的指数给定了范围[0...1000],那么可以使用间址技巧,将项的指数看成数组的下标,这样扫描一遍就可以得到结果。
 
#include <iostream>  
#include <fstream>  
#include <vector>  
#include <iomanip>  
using namespace std;  
  
  
struct item  
{  
    item(){}  
    item(int e, double c)  
    {  
        this->exp = e;  
        this->coef = c;  
    }  
    int exp;  
    double coef;  
};  
void merge(item *items1, item *items2, int a, int b);  
int main()  
{  
    //fstream cin("a.txt");  
  
    int a,b;  
    cin>>a;  
    int tmp = a;  
    item *items1 = new item[a];  
    while (tmp--)  
    {  
        int exp;  
        double coef;  
        cin>>exp>>coef;  
        item it(exp, coef);  
        items1[a - tmp - 1] = it;  
    }  
    cin>>b;  
    tmp = b;  
    item *items2 = new item[b];  
    while (tmp--)  
    {  
        int exp;  
        double coef;  
        cin>>exp>>coef;  
        item it(exp, coef);  
        items2[b - tmp - 1] = it;  
    }  
  
    merge(items1, items2, a, b);  
    return 0;  
}  
  
void merge(item *items1, item *items2, int a, int b)  
{  
    vector<item> v;  
    int i = 0,j = 0;  
    for(;i < a && j < b;)  
    {  
        if(items1[i].exp > items2[j].exp)  
        {  
            if(items1[i].coef != 0)  
                v.push_back(items1[i]);  
            i++;  
        }  
        else if(items1[i].exp < items2[j].exp)  
        {  
            if(items2[j].coef != 0)  
                v.push_back(items2[j]);  
            j++;  
        }  
        else  
        {  
            item it(items1[i].exp, items1[i].coef + items2[j].coef);  
            if (it.coef != 0)  
                v.push_back(it);  
            i++;  
            j++;  
        }  
    }  
  
    if(i == a)  
    {  
        while (j < b)  
        {  
            if(items2[j].coef != 0)  
                v.push_back(items2[j]);  
            j++;  
        }  
    }  
    else  
    {  
        while (i < a)  
        {  
            if(items1[i].coef != 0)  
                v.push_back(items1[i]);  
            i++;  
        }  
    }  
  
    cout<<v.size();  
    for (int i = 0; i < v.size(); i++)  
    {  
        cout<<" "<<v[i].exp<<" "<<fixed<<setprecision(1)<<v[i].coef;  
    }  
    cout<<endl;  
}  

 


补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,