清華機試oj——二叉樹遍歷

題目描述

編一個程序，讀入用戶輸入的一串先序遍歷字符串，根據此字符串建立一個二叉樹（以指針方式存儲）。例如如下的先序遍歷字符串：ABC##DE#G##F###其中“#”表示的是空格，空格字符代表空樹。建立起此二叉樹以後，再對二叉樹進行中序遍歷，輸出遍歷結果。

輸入描述:

輸入包括1行字符串，長度不超過100。

輸出描述:

可能有多組測試數據，對於每組數據，
輸出將輸入字符串建立二叉樹後中序遍歷的序列，每個字符後面都有一個空格。
每個輸出結果佔一行。

示例1

輸入

abc##de#g##f###

輸出

c b e g d f a 

解題思路

首先建立一個結構體用來存儲結點，其中包括結點的值（字符型）、左子樹（指針）、右子樹（指針），然後按照先序遍歷順序建樹（此時若爲空就是‘#’），最後中序遍歷輸出結果。

代碼實現

#include<bits/stdc++.h>
using namespace std;

string str;
int i;

struct TreeNode{
	char val;//結點的值（字符） 
	struct TreeNode *lchild, *rchild;//以指針方式存儲 
	TreeNode(char c):val(c), lchild(NULL), rchild(NULL){};
};

TreeNode* createTree(){	//按照前序建樹 
	char c = str[i++];
	if(c == '#') return NULL;
	TreeNode *root = new TreeNode(c);
	root->lchild = createTree();
	root->rchild = createTree();
	return root;
}

void inOrderTraversal(TreeNode* root){	//中序遍歷 
	if(!root) return;
	inOrderTraversal(root->lchild);	//左子樹 
	cout << root->val << " ";		//根節點 
	inOrderTraversal(root->rchild);	//右子樹 
}

int main(){
	while(cin >> str){
		i = 0;
		TreeNode *root = createTree();	//建樹 
		inOrderTraversal(root);			//中序遍歷 
		cout << endl;
	}
	return 0;
}

關於指針和二叉樹的應用一直是我的弱項，以後還需要多加總結和運用，這個代碼較符合模板形式，很有借鑑意義。

繼續加油吧(#^.^#)