HNU程序设计训练——相等的多项式

问题描述

小明现在在学习多项式的展开：就是把一个形如

（x+a1） （x+a2） ... （x+an）

展开成如下形式：

xn + b1xn-1 + b2xn-2 + ... + bn-1x + bn

比如 （x+1）（x+2）=x2 + 3x + 2

      (x+1)3 = x3 +3x2 +3x + 1

小明做了很多练习，但是不知道对错，现在请求你的帮助，判断小明的展开式是否正确。

输入格式

有多组测试数据。

每组测试数据有三行，第一行是一个正整数N，表示多项式最高指数。N=0表示输入结束，并且不需要处理。

第二行N个整数ai，用空格隔开，i=1，...,N(-100≤ai≤100)

第三行N个整数bi，用空格隔开，i=1，...,N，(-109≤bi≤109)

40%的测试数据 1 ≤ N < 5；

30%的测试数据 5 ≤ N < 10；

20%的测试数据10 ≤ N < 15；

10%的测试数据 15 ≤N≤ 20；

输出格式

　　对于每组测试数据，输出一行一个字符‘Y'如果展开式是正确的，输出‘N’如果展开式错误。

样例输入

2
1 2
3 2
3
1 1 1
3 3 1
4
0 0 0 1
0 0 0 1
0

样例输出

Y
Y
N

回溯法，解空间为排列树

#include <iostream>
using namespace std;

int N = 0;//N<=20
int a[21], b[21], c[21];
void fun(int n, int ci, int sum)
//目前所在排列树的层次，对应的选择常数的次数（a[i]相乘的次数），当前系数总和
{
	if (n == N)
	{
		if (ci == 0)//ci为0时对应展开的多项式中的x^n,系数为1,无需考虑
			return;
		else
		{
			c[ci] += sum;
			return;
		}
	}
	else
	{
		fun(n + 1, ci, sum);
		fun(n + 1, ci + 1, sum * a[n]);//选择当前的a[n],系数总和改变，对应的多项式次数+1
	}
}
void FLAG()
{
	for (int i = 1; i <= N; i++)
		if (b[i] != c[i])
		{
			printf("N\n");
			return;
		}
	printf("Y\n");//b[i]==c[i]均成立

}
int main()
{
	while (scanf("%d", &N), N)
	{
		for (int i = 0; i <= N; i++)
			c[i] = 0;//答案系数初始为0
		for (int i = 0; i < N; i++)
			scanf("%d", &a[i]);
		for (int i = 1; i <= N; i++)
			scanf("%d", &b[i]);//bn表示多项式次数是0，选择常数的次数是n

		fun(0, 0, 1);
		//第0层进入（0至n-1处理n次，在n==N时处理c[sum])，当前未进行选择，次数对应0，系数总和初始为1
		FLAG();//进入判断
	}
	return 0;
}