【PAT 甲级】1002 A+B for Polynomials (25分)（map）

1002 A+B for Polynomials (25分)

题目

给出两个多项式的系数和指数，求相加后的多项式系数和指数。

分析

用 map 存系数和指数的对应关系，将系数加到相同的指数里面即可。
注意只需要输出不为零的项。

#include <bits/stdc++.h>
using namespace std;
#define db(x) cout<<x<<endl

typedef long long ll;
const int INF = 0x3f3f3f3f;
const int N = 1e4 + 10;
const ll mod = 2147483648;

int n, m;
map<int, double, greater<int> > mp;
double ans[N];
int a; double b;

int main() {
	scanf("%d", &n);
	for (int i = 0; i < n; i++) {
		scanf("%d%lf", &a, &b);
		mp[a] += b;
	}
	scanf("%d", &m);
	for (int i = 0; i < m; i++) {
		scanf("%d%lf", &a, &b);
		mp[a] += b;
	}
	int cnt = 1;
	for (map<int, double>::iterator it = mp.begin(); it != mp.end(); it++) {
		if (it->second == 0) continue;
		ans[cnt++] = it->first;
		ans[cnt++] = it->second;
	}
	if (cnt == 1) {
		printf("0\n"); return 0;
	}
	printf("%d ", cnt >> 1);
	for (int i = 1; i < cnt; i += 2) {
		printf("%.0lf %.1lf%c", ans[i], ans[i + 1], " \n"[i==cnt-2]);
	}
	return 0;
}