PAT A1002 A+B for Polynomials(25)

题意

• 模拟多项式相加。

注意

1. 只输出非零项。
2. printf函数，控制输出小数位数%.1f

单词

1. polynomials 多项式
2. coefficient 系数

代码

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

void add(int k1, int k2, int n1[], int n2[], double a1[], double a2[])
{
	bool b1[10];
	bool b2[10];
	for (int bb = 0; bb < 10; bb++)
		b1[bb] = b2[bb] = false;
	int k3 = 0;
	int n3[20];
	double a3[1001];
	for (int i = k1 - 1; i >= 0; i--)
	{
		for (int j = k2 - 1; j >= 0; j--)
		{
			if (n1[i] == n2[j])   //由于是排序的，更快的做法是让每一步的j从上一次的j开始比
			{
				if ((a1[i] + a2[j]) != 0)             
				{
					n3[k3] = n1[i];
					a3[n3[k3++]] = a1[i] + a2[j];
				}
				b1[i] = b2[j] = true;
				break;
			}
		}
	}
	for (int i = k1 - 1; i >= 0; i--)
	{
		if (!b1[i])
		{
			n3[k3] = n1[i];
			a3[n3[k3++]] = a1[i];
		}
	}
	for (int i = k2 - 1; i >= 0; i--)
	{
		if (!b2[i])
		{
			n3[k3] = n2[i];
			a3[n3[k3++]] = a2[i];
		}
	}
	sort(n3, n3 + k3);                              
	printf("%d", k3);
	for (int r = k3 - 1; r >= 0; r--)
		printf(" %d %.1f", n3[r], a3[n3[r]]);   
}

int main()
{
	int n1[10], n2[10];
	double a1[10], a2[10];
	int k1, k2;
	cin >> k1;
	for (int i = k1; i >= 1; i--)
		cin >> n1[i - 1] >> a1[i - 1];
	cin >> k2;
	for (int i = k2; i >= 1; i--)
		cin >> n2[i - 1] >> a2[i - 1];
	add(k1, k2, n1, n2, a1, a2);
    return 0;
}

再贴一份吧，，把指数和系数这两个打包到一个结构体，然后定义排序函数。

（19年5月6日）给的两个多项式都是指数降序且指数不重复的，最多10项，所以能合并的项都是两个中各存在一个，先把这些找出来合并（如果合并为0则忽略）并且标记一下，然后再添加那些单独的项。
上下不同之处在于，上面是用了两个数组分别装指数和系数，因为指数给定了范围，所以系数是直接以对应指数为索引，最后直接对指数数组排序就行了。而下面是把指数系数一起打包为一个结构体，定义排序函数对指数排序就行了。（不用再开一个大小为1001的系数数组了）

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

struct st
{
	int e;
	double co;
}n3[20];                // no 1001   19.2.26

bool cmp(st s1, st s2)
{
	return s1.e < s2.e;
}

void add(int k1, int k2, int n1[], int n2[], double a1[], double a2[])
{
	bool b1[10];
	bool b2[10];
	for (int bb = 0; bb < 10; bb++)
		b1[bb] = b2[bb] = false;
	int k3 = 0;
	for (int i = k1 - 1; i >= 0; i--)
	{
		for (int j = k2 - 1; j >= 0; j--)
		{
			if (n1[i] == n2[j])
			{
				if ((a1[i] + a2[j]) != 0)             
				{
					n3[k3].e = n1[i];
					n3[k3++].co = a1[i] + a2[j];
				}
				b1[i] = b2[j] = true;
				break;                                17.2.10
			}
		}
	}
	for (int i = k1 - 1; i >= 0; i--)
	{
		if (!b1[i])
		{
			n3[k3].e = n1[i];
			n3[k3++].co = a1[i];
		}
	}
	for (int i = k2 - 1; i >= 0; i--)
	{
		if (!b2[i])
		{
			n3[k3].e = n2[i];
			n3[k3++].co = a2[i];
		}
	}
	//sort(n3, n3 + k3);                          
	sort(n3, n3 + k3, cmp);
	printf("%d", k3);
	for (int r = k3 - 1; r >= 0; r--)
		printf(" %d %.1f", n3[r].e, n3[r].co);   
}

int main()
{
	int n1[10], n2[10];
	double a1[10], a2[10];
	int k1, k2;
	cin >> k1;
	for (int i = k1; i >= 1; i--)
		cin >> n1[i - 1] >> a1[i - 1];
	cin >> k2;
	for (int i = k2; i >= 1; i--)
		cin >> n2[i - 1] >> a2[i - 1];
	add(k1, k2, n1, n2, a1, a2);
    return 0;
}