L2-011. 玩转二叉树

            L2-011. 玩转二叉树

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（<=30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
1 2 3 4 5 6 7
4 1 3 2 6 5 7

输出样例：

4 6 1 7 5 3 2

---

提交代码

#include<iostream>
#include<string.h>
#include<algorithm>
#include<queue> 
using namespace std;
//首先要明白输出该树的反转后的层次遍历就是在层次遍历时，先输出右子树，在输出左子树。。 
int pre[200],in[200],n; 
struct node
{
	int data;
	node *ld;
	node *rd;
};
node *create(int ps,int pe,int is,int ie)
{
	if(ps>pe) return NULL;
	
	node *r=new node;
	r->data=pre[ps];
	
	int p;
	for(p=is;p<=ie;p++)
	{
		if(in[p]==pre[ps])
			break;
	}
	int k=p-is;//k=3
	r->ld=create(ps+1,ps+k,is,p-1);
	r->rd=create(ps+k+1,pe,p+1,ie);
	return r;
}
int sum=0;
int bfs(node *root)  
{  
    queue<node*>q;  
    node *temp;  
    q.push(root);  
    while(!q.empty())  
    {  
        temp=q.front();  
        q.pop(); 
		if(temp->data!=0) 
          cout<<temp->data; //不知为何会输出0,所以用这两天语句控制不输出零 
		else
			continue; //等于0的时候跳过 
        sum++;  
        if(sum<n) cout<<" ";//控制格式 
		if(temp->rd!=NULL) q.push(temp->rd);  //反转输出 
        if(temp->ld!=NULL) q.push(temp->ld);        
    }  
}  
int main()
{
	cin>>n;
	for(int i=0;i<n;i++)
		cin>>in[i];
	for(int i=0;i<n;i++)
		cin>>pre[i];
	node *root=create(0,n+1,0,n+1);
    bfs(root);
    //cout<<"ok";
}

#include<iostream>
#include<math.h>
#include<memory.h>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int pre[100],in[100],n;
struct node
{
	int data;
	node *ld;
	node *rd;
};
node *create(int ps,int pe,int is,int ie)
{
	if(ps>pe) return NULL;
	node *b=new node;
	b->data=pre[ps];
	int k;
	for(k=is;k<=ie;k++)
	{
		if(in[k]==pre[ps])
			break;
	}
	int num=k-is;
	b->ld=create(ps+1,ps+num,is,k-1);
	b->rd=create(ps+num+1,pe,k+1,ie);
	return b;
}
int sum=0;
void dfs(node *r)
{
	queue<node*>q;
	q.push(r);
	node *temp;
	while(!q.empty())
	{	sum++;
		temp=q.front();
		q.pop();
		cout<<temp->data;
		if(sum<n) cout<<" ";
		if(temp->rd!=NULL) q.push(temp->rd);
		if(temp->ld!=NULL) q.push(temp->ld);		
	}
}
int main()
{
	int i;
	cin>>n;
	for(i=0;i<n;i++)
		cin>>in[i];
	for(i=0;i<n;i++)
		cin>>pre[i];
	node *root;
	root=create(0,n-1,0,n-1);
	
	dfs(root);
}