相等的多项式

题目描述

问题描述

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

（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 <bits/stdc++.h>
using namespace std;
int main(){
	int n;
	while(cin >> n && n != 0){
		map<int, int> Ming, Me;
		int a[n], tmp, flag = 1;
		for(int i = 0; i < n; i++)
			cin >> a[i];
		for(int i = 0; i < n; i++){
			cin >> tmp;
			Ming[n - i - 1] = tmp;
			Me[n - i - 1] = 0;
		}
		Me[1] = 1;
		Me[0] = a[0];
		for(int i = 1; i < n; i++){
			for(int j = n - 1; j >= 0; j--){
				Me[j] *= a[i];
				Me[j] += Me[j - 1];
			}
		}
		for(int j = n - 1; j >= 0; j--)
			if(Me[j] != Ming[j]){
				flag = 0;
				break;
			}		
		if(flag == 1) cout << "Y\n";
		else cout << "N\n";
	}
	return 0;
} 

如果对各位看官有帮助不妨留下一个点赞￣ω￣=。