树的层次遍历
例题描述
输入一棵二叉树,你的任务是按照从上到下,从左到右的顺序输出各个结点的值,每个节点都按照从根节点到它的移动序列给出(L表示左,R表示右)。在输入中,每个结点的左括号和右括号之间没有空格,相邻接点之间用一个空格隔开。每棵树的输入用一对空括号“()”结束,这个空括号本身不代表一个结点。注意,如果从根到某个叶结点的路径上有的结点没有在输入中给出,或者给出超过一次,应当输出-1.结点个数不超过256.
样例输入
(11,LL) (7,LLL) (8,R) (5,) (4,L) (13,RL) (2,LLR) (1,RRR) (4,RR) ()
(3,L) (4,R) ()
样例输出
5 4 8 11 13 4 7 2 1
-1
利用指针实现二叉树
题记学完数据结构应该很容易想到直接用指针创建二叉树,但是要注意内存泄漏,代码中有注释可以帮助理解。另外,利用队列实现广度优先遍历应该掌握。
代码如下
#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
#define maxn 20
using namespace std;
// 用指针实现二叉树
char s[maxn]; //输入当前节点的路径
vector<int>ans; //记录输出数据
int cnt; //记录节点数
struct Node{
bool have_value; //是否已经赋值
int v; //结点的值
Node *left,*right; //左子树,右子树
Node():have_value(false),left(NULL),right(NULL){}//构造函数
};
Node *root;//根节点
bool failed;//记录输入是否有误,如果有误则为true
Node *newnode(){return new Node();}
void remove_tree(Node* u){ //深度优先遍历二叉树,从深处释放内存
if(u==NULL)
return;
remove_tree(u->left);
remove_tree(u->right);
delete u;
}
void addnode(int v,char *s){ //根据字符串s中的指令添加值为v的新结点
int n=strlen(s);
Node* u=root;
for(int i=0;i<n;i++)
if(s[i]=='L'){
if(u->left==NULL)
u->left=newnode();
u=u->left;
}
else if (s[i]=='R') {
if(u->right==NULL)
u->right=newnode();
u=u->right;
}
if(u->have_value)
failed=true; //如果这个结点已经被赋过值,那么输入有误
u->v=v;
u->have_value=true;
}
bool read_input(){
failed=false;
remove_tree(root); //内存泄漏对平常的题没啥影响,但是这样做更能体现专业性
root=newnode();
cnt=0;
for (;;) {
if(scanf("%s",s)!=1)
return false;
if(!strcmp(s,"()"))
break;
int v;
sscanf(&s[1],"%d",&v); //读取插入节点的值
addnode(v,strchr(s,',')+1);//找第一次出现逗号的位置,下一个位置就是第一条路径方向
cnt++;
}
return true;
}
bool bfs(vector<int >&ans){ //利用队列实现广度优先遍历
queue<Node*>q;
ans.clear(); //清除上一组数据的vector
q.push(root);
while(!q.empty()){
Node* u=q.front();
q.pop();
if(!u->have_value) //经过的这个点没有赋值,说明输入错误
return false;
ans.push_back(u->v);
if(u->left!=NULL)
q.push(u->left);
if(u->right!=NULL)
q.push(u->right);
}
return true;
}
int main()
{
while(read_input()){
if(!failed &&bfs(ans)){
for(int i=0;i<cnt;i++)
printf("%d ",ans[i]);
}
else
printf("%d",-1);
printf("\n");
}
return 0;
}