pat 甲级 A1002 A+B for Polynomials (25分)

题目链接：

https://pintia.cn/problem-sets/994805342720868352/problems/994805526272000000

 

题目大意：

计算两个多项式的和，多项式表示方式如下：

K N​1​​ a​N​1​​​​ N​2​​ a​N​2​​​​ ... N​K​​ a​N​K​​​​ ，1≤K≤10，0≤N​K​​<⋯<N​2​​<N​1​​≤1000.

首先，给出多项式的项数K，然后以指数递减给出K个单项式的系数和指数。

输入两个多项式，并且输出多项式之和，格式如上面所示。

 

思路分析：

先获得第一个多项式的指数和系数，然后在输入第二个系数以及指数时，将指数大者的系数保存到相应的结果数组中，若对应的指数相同，则把系数相加，再判断是否为0，考虑将其保存到结果数组中。

注意，在内层循环中，需要判断ex==A_ex[j]，若相同，则需执行内循环，且退出循环后，还需执行内循环后的语句，导致此时结果数组中存在多余的输入系数，所以添加flag用于判断内循环中，满足ex==A_ex[j]条件的语句是否执行。

参考代码：

#include<cstdio>
const int maxn=20;
int A_ex[maxn],B_ex[maxn];//指数
double A_co[maxn],B_co[maxn];//系数

int main(){
  int k1,k2,i,j = 0,ex,k = 0;
  scanf("%d",&k1);
  for(i=0;i<k1;i++){
    scanf("%d",&A_ex[i]);
    scanf("%lf",&A_co[i]);
  }
  double cof;
  scanf("%d",&k2);
  for(i=0;i<k2;i++){
    scanf("%d %lf",&ex, &cof);
    bool flag = true;
    for(; j < k1 && A_ex[j] >= ex; j++){
        if(ex ==A_ex[j]){
            flag = false;
        B_ex[k]=ex;
        B_co[k] = A_co[j] + cof;
        if(B_co[k]) k++;
    }else{
        B_ex[k]= A_ex[j];
        B_co[k]= A_co[j];k++;}
    }
    if(flag){
    B_ex[k]=ex;
    B_co[k]=cof;
      k++;}
  }
  for(;j < k1;j++,k++){
       B_ex[k]=A_ex[j];
       B_co[k]=A_co[j];}
  printf("%d",k);
  for(i=0;i<k;i++){
    printf(" %d %.1f",B_ex[i],B_co[i]);
  }
  return 0;
}

 

 

对于指数不大的情况，可以利用空间换时间的方法。QAQ而且代码简单，明了。

#include <cstdio>
double ans[1010];
int exp, k, num = 0;   
double cof;

int main(){
    for(int j = 0; j < 2; j++){
        scanf("%d", &k);
        for(int i = 0; i < k; i++){
            scanf("%d %lf", &exp, &cof);
            ans[exp] += cof;
        }
    }
    for(int i = 0; i < 1010; i++)
        if(ans[i]) num++;
    printf("%d",num);
    for(int i = 1009; i >= 0; i--){
        if(ans[i]){
            printf(" %d %.1f", i, ans[i]);
        }
    }
}