中国大学MOOC--《数据结构》——02-线性结构2 一元多项式的乘法与加法运算【解法一：静态数组法】

这题有两种解法，另一解法请参考：https://blog.csdn.net/garrulousabyss/article/details/88327135

题干：

设计函数分别求两个一元多项式的乘积与和。

输入格式:

输入分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

我的代码：

 

#include <iostream>
#include <vector>
using namespace std;

void add(int N1,int * coef1, int * power1, int N2,int* coef2, int * power2)
{
  int coef_result[2001]={0};
  vector<int>v;
  //0 power is 1000 index
  for(int i=0;i<N1;i++)
    coef_result[power1[i]+1000] += coef1[i];
  for(int i=0;i<N2;i++)
    coef_result[power2[i]+1000] += coef2[i];
  //output
  for(int i=0;i<2001;i++)
  {
    if(coef_result[i] != 0){
      v.push_back(i-1000);
      v.push_back(coef_result[i]);
    }
  }

  int size = v.size();
  if(size==0)
    cout<<0<<" "<<0;
  else{
      for(int i=size-1;i>=0;i--)
        {
            cout<<v[i];
            if(i!=0)
            cout<<" ";
        }
  }

}

void multi(int N1,int * coef1, int * power1, int N2,int* coef2, int * power2)
{
  int coef_result[4001]={0};
  vector<int>v;

  for(int i=0;i<N1;i++)
  {
    for(int j=0;j<N2;j++)
    {
      coef_result[power1[i]+power2[j]+2000] += coef1[i]*coef2[j];
    }
  }

  //output
  for(int i=0;i<4001;i++)
  {
    if(coef_result[i] != 0){
      v.push_back(i-2000);
      v.push_back(coef_result[i]);
    }
  }

  int size = v.size();
  if(size==0)
    cout<<0<<" "<<0;
  else{
      for(int i=size-1;i>=0;i--)
        {
            cout<<v[i];
            if(i!=0)
            cout<<" ";
        }
  }
}

int main()
{
  int N1,N2;
  cin>>N1;
  int coef1[N1],power1[N1];
  for(int i=0;i<N1;i++)
  {
    cin>>coef1[i]>>power1[i];
  }

  cin>>N2;
  int coef2[N2],power2[N2];
  for(int i=0;i<N2;i++)
  {
    cin>>coef2[i]>>power2[i];
  }


  multi(N1,coef1,power1,N2,coef2,power2);
  cout<<endl;
  add(N1,coef1,power1,N2,coef2,power2);

  return 0;
}