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

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;
} 
  • 4
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值