PTA 7-1 玩转二叉树

https://pintia.cn/problem-sets/1259041418971803648/problems/1259046259492220928
(题目链接如上)
题意:
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。
利用中序遍历和前序遍历建树:

node* buildPreInTree(int pl,int pr,int il,int ir){
	if(il>ir) return NULL;
	node *root=new node;
	root->num=pre[pl];
	int mid;
	for(int i=il;i<=ir;i++){
		if(pre[pl]==in[i]){
			mid=i;break;
		}
	}
	int llen=mid-il,rlen=ir-mid;
	root->left=buildPreInTree(pl+1,pl+llen,il,mid-1);
	root->right=buildPreInTree(pl+llen+1,pr,mid+1,ir);
	return root;
}

这道题中,要将root->left和root->right换位,来建树。
二叉树的层序遍历:

int levelOrderTraverse(node* rt){
	queue<node*>q;
	q.push(rt);
	int k=0;
	while(!q.empty()){
		node* root=q.front();
		q.pop();
		lev[k++]=root->num;
		if(root->left!=NULL) q.push(root->left);
		if(root->right!=NULL) q.push(root->right);
	}
	return k;
}

利用队列,考虑到输出格式,就先把它存储在lev数组里了。
AC代码:

#include <iostream> 
#include <queue>
using namespace std;
int n,pre[35],in[35],lev[35];
struct node{
	int num;
	node *left,*right;
};
node* buildPreInTree(int pl,int pr,int il,int ir){
	if(il>ir) return NULL;
	node *root=new node;
	root->num=pre[pl];
	int mid;
	for(int i=il;i<=ir;i++){
		if(pre[pl]==in[i]){
			mid=i;break;
		}
	}
	int llen=mid-il,rlen=ir-mid;
	root->right=buildPreInTree(pl+1,pl+llen,il,mid-1);
	root->left=buildPreInTree(pl+llen+1,pr,mid+1,ir);
	return root;
}
int levelOrderTraverse(node* rt){
	queue<node*>q;
	q.push(rt);
	int k=0;
	while(!q.empty()){
		node* root=q.front();
		q.pop();
		lev[k++]=root->num;
		if(root->left!=NULL) q.push(root->left);
		if(root->right!=NULL) q.push(root->right);
	}
	return k;
}
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 *rt;
	rt=buildPreInTree(0,n-1,0,n-1);
	int k;
	k=levelOrderTraverse(rt);
	for(int i=0;i<k;i++){
		if(i==0) cout<<lev[i];
		else cout<<" "<<lev[i];
	}
	cout<<endl;
	return 0;
}
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 1024 设计师: 上身试试
应支付0元
点击重新获取
扫码支付

支付成功即可阅读