算法与数据结构7-2 一元多项式的乘法与加法运算 (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

本应该用链表做的，但是太复杂了，索性用数组。

#include<iostream>
#include<set>
using namespace std;
const int maxn = 1e5;
int a[maxn], b[maxn],c[maxn],d[maxn];

int main() {
	int i, j;
	int x, z;
	int cnt = 0;
	cin >> i;
	while (i--) {
		//scanf("%d %d", &x, &z);
		cin >> x >> z;
		a[z] += x;
	}
	cin >> i;
	while (i--) {
		//scanf("%d %d", &x, &z);
		cin >> x >> z;
		b[z] += x;
	}
	for (i = maxn - 1; i >= 0; i--) {
		if (a[i]) {
			for (j = 0; j < maxn; j++) {
				if (b[j]) c[i + j] += a[i] * b[j];
			}
		}
	} 
	for (i = maxn - 1; i >= 0; i--) {
		if (c[i]) {
			if (cnt) cout << " ";
			printf("%d %d", c[i], i);
			cnt++;
		}
	}

	if (!cnt) printf("0 0");
	cout << endl;
	for (i = maxn - 1; i >= 0; i--)
		if (a[i]) d[i] += a[i];
	for (j = maxn - 1; j >= 0; j--)
		if (b[j]) d[j] += b[j];
	cnt = 0;
	for (i = maxn - 1; i >= 0; i--) {
		if (d[i]) {
			if (cnt) cout << " ";
			printf("%d %d", d[i], i);
			cnt++;
		}
	}
	if (!cnt) printf("0 0");
	system("pause");
	return 0;
}