一元多项式的乘法与加法运算(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
思路
尽管这道题是操练链表的题,但我还是想用方便的容器来实现一元多项式。在做乘法的时候用到了find,写的时候感觉会最终在这点超时,结果没有,开心地过了。在纸头上用笔自己模拟一遍就能写出代码。加法用的vector,乘法因为一项一项会变乱,所以用set保持排列,具体细节见代码。这应该是陈越数据结构mooc的课后作业,乘法和加法一起写的话代码还是有点长度的,140多行呢。
AC源码
/*
Name:7-3 一元多项式的乘法与加法运算(20 分)
Author: spencercjh
Date: 25/09/17 08:19
Description: 数据结构任务一
*/
#include<iostream>
#include<vector>
#include<set>
using namespace::std;
struct node
{
int a;
int exp;
bool operator<(const node& a) const
{
return exp>a.exp;
}
bool operator==(const node& a)const
{
return exp==a.exp;
}
};
void add(vector<node> p1,vector<node> p2)
{
int N=p1.size()+p2.size();
vector<node> ans;
for(int i=0,j=0;i<min(p1.size(),p2.size());)
{
if(p1[i].exp>p2[j].exp) 每次把指数(exponent)较大的加入链表fans
{
ans.push_back(p1[i]);
i++;
}
else if(p1[i].exp<p2[j].exp)
{
ans.push_back(p2[j]);
j++;
}
else
{
node temp;
temp.a=p1[i].a+p2[j].a;
temp.exp=p1[i].exp;
if(temp.a==0)
{
i++;
j++;
}
else
{
ans.push_back(temp);
i++;
j++;
}
}
}
if(p1.size()>p2.size())
for(int i=min(p1.size(),p2.size());i<p1.size();i++)
ans.push_back(p1[i]);
else if(p1.size()<p2.size())
for(int i=min(p1.size(),p2.size());i<p2.size();i++)
ans.push_back(p2[i]);
if(ans.size()==0)
{
printf("0 0\n");
return;
}
for(int i=0;i<ans.size();i++)
if(i!=ans.size()-1)
printf("%d %d ",ans[i].a,ans[i].exp);
else
printf("%d %d",ans[i].a,ans[i].exp);
cout<<endl;
}
void mutiplication(vector<node> p1,vector<node> p2)
{
set<node> ans;
for(int i=0;i<p2.size();i++)
{
for(int j=0;j<p1.size();j++)
{
node temp;
temp.a=p1[j].a*p2[i].a;
temp.exp=p1[j].exp+p2[i].exp;
auto it=ans.find(temp);//已经重载过<和==,find找出的是指数相同的项,而不是真正完全相同的项
if(it!=ans.end())//找到了指数相同的项,系数相加
{
temp.a+=it->a;
ans.erase(*it);//经过我几次测试,发现要把找到的这一项删掉,否则影响后面的插入。可能是因为已经存在指数为it->exp的项了,而我又重载过<和==,内部判定我前后插入了一模一样的东西,导致了错误
}
if(temp.a==0)
continue;
else
ans.insert(temp);
}
}
auto it=ans.begin();
printf("%d %d",it->a,it->exp);
it++;
for(;it!=ans.end();it++)
printf(" %d %d",it->a,it->exp);
cout<<endl;
}
int main()
{
int n1;
scanf("%d",&n1);
vector<node> p1(n1);
for(int i=0;i<n1;i++)
scanf("%d %d",&p1[i].a,&p1[i].exp);
int n2;
scanf("%d",&n2);
vector<node> p2(n2);
for(int i=0;i<n2;i++)
scanf("%d %d",&p2[i].a,&p2[i].exp);
if(n1==0||n2==0)
{
printf("0 0\n");
if(n1==0)
for(int i=0;i<n2;i++)
{
if(i!=n2-1)
printf("%d %d ",p2[i].a,p2[i].exp);
else
printf("%d %d",p2[i].a,p2[i].exp);
}
if(n2==0)
for(int i=0;i<n1;i++)
{
if(i!=n2-1)
printf("%d %d ",p1[i].a,p1[i].exp);
else
printf("%d %d",p1[i].a,p1[i].exp);
}
return 0;
}
mutiplication(p1,p2);
add(p1,p2);
}