算法竞赛入门经典6.3.2层次遍历

算法竞赛入门经典6.3.2层次遍历

输入一颗二叉树，你的任务是按从上到下，从左到右的顺序输出各个节点的值。每个节点都按照从根节点到它的移动序列给出（L表示左，R表示右）。在输入中，每个节点的左括号之间没有空格，相邻节点之间用一个空格隔开。每棵树的输入用一对空括号（）结束（这对括号本身不代表一个结点），如图所示。

5

4 8

11 13 4

7 2 1

注意，如果从根到某个叶节点的路径上有的结点没有再输入中给出，或者给出了超过一次，应当输出-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>

using namespace std;

const int MAXN=256;

char s[MAXN+10];

int n,ans[MAXN];

int failed;

typedef struct Tnode

{

int have_value;

int v;

struct Tnode* left,*right;

}Node;

Node* root;

Node* newnode()

{

Node* u=(Node*) malloc(sizeof(Node));

if(u!=NULL)

{

u->have_value=0;

u->left=u->right=NULL;

}

return u;

}

void addnode(int v,char* s)

{

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=1;

u->v=v;

u->have_value=1;

}

void remove_tree(Node* u)

{

if(u==NULL)

return;

remove_tree(u->left);

remove_tree(u->right);

free(u);

}

int read_input()

{

failed=0;

remove_tree(root);

root=newnode();

for(;;)

{

if(scanf("%s",s)!=1)

return 0;

if(!strcmp(s,"()"))

break;

int v;

sscanf(&s[1],"%d",&v);

addnode(v,strchr(s,',')+1);

}

return 1;

}

int bfs()

{

int front=0;

int rear=1;

Node* q[MAXN];

q[0]=root;

while(front<rear)

{

Node* u=q[front++];

if(!u->have_value)

return 0;

ans[n++]=u->v;

if(u->left!=NULL)

q[rear++]=u->left;

if(u->right!=NULL)

q[rear++]=u->right;

}

return 1;

}

int main()

{

read_input();

bfs();

if(failed==1)

cout<<"-1"<<endl;

else

for(int i=0;i<n;++i)

cout<<ans[i]<<" ";

system("pause");

return 0;

}