PTA 多项式A除以B

171 阅读2分钟

这仍然是一道关于A/B的题,只不过A和B都换成了多项式。你需要计算两个多项式相除的商Q和余R,其中R的阶数必须小于B的阶数。

输入格式:

输入分两行,每行给出一个非零多项式,先给出A,再给出B。每行的格式如下:

N e[1] c[1] ... e[N] c[N]

其中N是该多项式非零项的个数,e[i]是第i个非零项的指数,c[i]是第i个非零项的系数。各项按照指数递减的顺序给出,保证所有指数是各不相同的非负整数,所有系数是非零整数,所有整数在整型范围内。

输出格式:

分两行先后输出商和余,输出格式与输入格式相同,输出的系数保留小数点后1位。同行数字间以1个空格分隔,行首尾不得有多余空格。注意:零多项式是一个特殊多项式,对应输出为0 0 0.0。但非零多项式不能输出零系数(包括舍入后为0.0)的项。在样例中,余多项式其实有常数项-1/27,但因其舍入后为0.0,故不输出。

输入样例:

4 4 1 2 -3 1 -1 0 -1
3 2 3 1 -2 0 1
结尾无空行

输出样例:

3 2 0.3 1 0.2 0 -1.0
1 1 -3.1
结尾无空行

代码:

#include <bits/stdc++.h>
using namespace std;
map<int,double>mp;
map<int,double>mp1;
struct ListNode
{
    int e;
    double c;
}b[100000];
int main()
{
    int n,m,max,e,dis;
    int ans1=0,ans2=0;
    double c;
    cin >> n;
    mp.clear();
    mp1.clear();
    for(int i=1; i <= n; i++ ) 
    {
        cin >> e >> c;
        mp[e]=mp[e]+c;
        if(i==1)
        {
            max=e;
        }
    }
    cin >> m;
    for(int i=0;i<m;i++)
    {
        cin >> b[i].e >> b[i].c;  
    }
    for(int i=max;i>=b[0].e;i--)
    {
        dis=i-b[0].e;
        mp1[dis]=mp[i]/b[0].c;
        for(int j=0; j<m;j++)
        {
            mp[dis+b[j].e]=mp[dis+b[j].e]-mp1[dis]*b[j].c;
        }
    }
    for(int i=max;i>=0;i--)
    {
        if(fabs(mp[i])<5e-2)
        {
            mp[i]=0;
        }
        if(fabs(mp1[i])<5e-2)
        {
            mp1[i]=0;
        }
        if(mp1[i]!=0)
        {
            ans1++;
        }
        if(mp[i]!=0)
        {
            ans2++;
        }
    }
    cout << ans1;
    for(int i=max;i>=0;i--)
    {
        if(mp1[i]!=0)
        {
            printf(" %d %.1lf",i,mp1[i]);
        }
    }
    if(ans1==0)
    {
        printf(" 0 0.0");
    }
    cout << endl;
    cout << ans2;
    for(int i=max;i>=0;i--)
    {
        if(mp[i]!=0)
        {
            printf(" %d %.1lf",i,mp[i]);
        }
    }
    if(ans2==0)
    {
        printf(" 0 0.0");
    }
    cout << endl;
    return 0;
}

提交结果: