算法竞赛入门经典 数组实现树的层序遍历的一个小错误

Posted 2022-12-12 As_zyh

p154 紫书原文：

纠正一下这里的newnode()函数中，have_value[root] = false 应该是 have_value[u] = false

附上代码，不知道原题的可以上网搜一下 例题6_7 Uva122
树的层序遍历数组(静态链表)实现:

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;

const int MAX_N = 256 + 10;
char s[MAX_N];                                    //保存读入结点 
bool failed;

const int root = 1;
int cnt;
int left[MAX_N], right[MAX_N], value[MAX_N];
bool have_value[MAX_N];

void newtree() 
	left[root] = right[root] = 0;
	have_value[root] = false;
	cnt = root;


int newnode() 
	int u = ++cnt;
	left[u] = right[u] = 0;
	have_value[u] = false;
	return u;


void addnode(int v, char* s) 
	int n = strlen(s);
	int u = root;                                    //从根结点开始往下走
	for(int i = 0; i < n; i++)
		if(s[i] == 'L') 
			if(left[u] == 0) left[u] = newnode();   //结点不存在，建立新结点
			u = left[u];                               //往左走 
		 else if(s[i] == 'R') 
			if(right[u] == 0) right[u] = newnode();
			u = right[u];
		                                        //忽略其它情况，即最后那个多余的右括号
	if(have_value[u]) failed = true;         //已经赋过值，表明输入有误
	value[u] = v;
	have_value[u] = true;                        //别忘记做标记       


bool read_input() 
	failed = false;
	newtree();                                  //创建根结点                           
	for(;;) 
		if(scanf("%s", s) != 1) return false;   //整个输入结束 
		if(!strcmp(s, "()")) break;             //读到结束标志，退出循环
		int v;
		sscanf(&s[1], "%d", &v);                //读入结点值
		addnode(v, strchr(s, ',')+1);           //查找逗号，然后插入结点 
	
	return true;


bool bfs(vector<int>& ans) 
	queue<int> q;
	ans.clear();
	q.push(root);                                //初始时只有一个根结点
	while(!q.empty()) 
		int u = q.front(); q.pop();            
		if(!have_value[u]) return false;         //有结点没有被赋值过，表明输入有误
		ans.push_back(value[u]);                     //增加到输出序列尾部
		if(left[u] != 0) q.push(left[u]);     //把左子节点（如果有）放进队列
		if(right[u] != 0) q.push(right[u]);   //把右子节点（如果有）放进队列 
	 
	return true;                                 //输入正确 


int main()
	#ifdef LOCAL
	freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);
	#endif
	while(read_input()) 
		vector<int> ans;
		if(!bfs(ans)) failed = true;
		if(!failed) 
			for(int i = 0; i < ans.size(); i++) 
				if(i > 0) printf(" ");
				printf("%d", ans[i]);
			
			printf("\\n");
		 else 
			printf("not complete\\n");
		
	
	return 0;
 

树的层序遍历链表（指针）实现:

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<iostream>
using namespace std;

const int MAX_N = 256 + 10;
char s[MAX_N];                                  //保存读入结点 
bool failed;

//结点类型
struct Node
	bool have_value;                            //是否被赋值过
	int v;                                      //结点值
	Node *left, *right;
	Node():have_value(false),left(NULL),right(NULL)    //构造函数 
; 

Node* root;                                     //二叉树的根结点

Node * newnode()  return new Node();  

void remove_tree(Node* u) 
	if(u == NULL) return;                        //提前判断比较稳妥
	remove_tree(u->left);                        //递归释放左子树空间
	remove_tree(u->right);                       //递归释放右子树空间
	delete u;                                    //调用u的析构函数并释放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 = true;         //已经赋过值，表明输入有误
		u->v = v;
		u->have_value = true;                    //别忘记做标记       


bool read_input() 
	failed = false;
	remove_tree(root);
	root = newnode();                           //创建根结点 
	for(;;) 
		if(scanf("%s", s) != 1) return false;   //整个输入结束 
		if(!strcmp(s, "()")) break;             //读到结束标志，退出循环
		int v;
		sscanf(&s[1], "%d", &v);                //读入结点值
		addnode(v, strchr(s, ',')+1);           //查找逗号，然后插入结点 
	
	return true;


bool bfs(vector<int>& ans) 
	queue<Node*> q;
	ans.clear();
	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()
	#ifdef LOCAL
	freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);
	#endif
	while(read_input()) 
		vector<int> ans;
		if(!bfs(ans)) failed = true;
		if(!failed) 
			for(int i = 0; i < ans.size(); i++) 
				if(i > 0) printf(" ");
				printf("%d", ans[i]);
			
			printf("\\n");
		 else 
			printf("not complete\\n");
		
	
	return 0;