一元多项式的乘法与加法运算（模拟）

题意

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

输入格式:

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

思路

$dig(b)$ 表示多项式中指数为 $b$ 的项的系数，比如 $ax^b$ 就可以表示为 $dig(b) = a$ 。那么直接模拟这个多项式乘法和加法计算过程即可。

AC代码

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
typedef pair<int, int> pii;
const int maxn = 4000+40;
const int inf = 2000+5;
int dig[2][maxn];
pii f[2][maxn];

void solve(int n, int m) {
    int a, b, x, y;
    for(int i = 0; i < n; i++) {
        a = f[0][i].first, x = f[0][i].second;
        for(int j = 0; j < m; j++) {
            b = f[1][j].first, y = f[1][j].second;
            dig[0][x+y+inf] += a*b;
        }
    }
}

void print(int *a) {
    vector<pii> ret;
    int tol = 0;
    for(int i = maxn-1; i >= 0; i--) {
        if(a[i] != 0) {
            ret.push_back(make_pair(a[i], i-inf));
            tol++;
        }
    }
    if(tol == 0) {
        printf("0 0\n");
    } else {
        int len = ret.size();
        for(int i = 0; i < len; i++) {
            printf("%d %d%c", ret[i].first, ret[i].second, i==len-1?'\n':' ');
        }
    }
}

int main() {
    int n, m;
    while(scanf("%d", &n) == 1) {
        memset(dig, 0, sizeof(dig));
        int a, b;
        for(int i = 0; i < n; i++) {
            scanf("%d%d", &f[0][i].first, &f[0][i].second);
            a = f[0][i].first, b = f[0][i].second;
            dig[1][b+inf] += a;
        }
        scanf("%d", &m);
        for(int i = 0; i < m; i++)  {
            scanf("%d%d", &f[1][i].first, &f[1][i].second);
            a = f[1][i].first, b = f[1][i].second;
            dig[1][b+inf] += a;
        }
        solve(n, m);
        print(dig[0]);
        print(dig[1]);
    }
    return 0;
}

如有不当之处欢迎指出！

一元多项式的乘法与加法运算 （模拟）

题意