【C++编程题】相等的多项式（回溯法，排列树）

【问题描述】

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

（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

【思路】

这道题貌似暴力也可以做,几个for循环嵌套...这里用的是回溯法

回溯法就是要构建【排列树】或者【子集树】，这里本质是模拟真实的多项式相乘；

如：（x+a1）（x+a2）(x+a3)（x+a4）

得到的系数b1,b2,b3,b4。如果一个a1被选中，对应位置为1；如果不选它，则是0

那么任意只挑选一个a，就有：1000，0100，0010，0001四种  把他们对应的值加起来就是b1；

同理，任意选取2个a：0110，0011，1100，1010，1001 ，对应的排列情况的值相加就是b2...

看到这里应该明白了，就是一个穷尽排列的过程；

一棵二叉树，左走为0，右走为1，到达底层时，就得到了所有的排列...

【AC代码】

  #include<string>
 #include<iostream>
 #include<sstream>
 #include<string.h>
 #include<vector>
 #include<cmath>
 using namespace std;
 
 int a[30];
 int b[30];
 int res[30];  //答案数组，res[i]表示取出任意i个所得到的值
 int n;
 //0~n-1层  第n层为结果层 
 void sour(int now_floor,int num_1,int sum){//开始层0，被选取的a的个数，该路径选取的a的乘积
 	
    if(now_floor==n){     //任意的排列结果都在结果层之中了 
 		res[num_1]+=sum;
 		return;
	 } 
	 
	sour(now_floor+1,num_1,sum);                 //往左走，不选为0
	sour(now_floor+1,num_1+1,sum*a[now_floor]);	 //往右走，选择为1，sum+1，同时乘积改变
 }
 
 int main(){
 
 	while(cin>>n && n){
 		
 		memset(res,0,sizeof(res));
 		
 		for(int i=0;i<n;i++) cin>>a[i];
 		for(int i=0;i<n;i++) cin>>b[i];
		
		sour(0,0,1);  
		
		//for(int i=1;i<=n;i++) cout<<res[i]<<" ";
		
		int flag=1;
		for(int i=0;i<n;i++){      //对比结果数组和b数组
			if(res[i+1]!=b[i]){
				flag=0;
				break;
			}
		}
		if(flag) cout<<"Y"<<endl;
		else cout<<"N"<<endl;
		
	 }
 	
 }

【写在后面】 

觉得对你有帮助的话记得一键三连哦~（误

相关题目放在了【HNU CJ】专栏

如果有啥问题可以评论留言或者私我哦~