1. 程式人生 > >二叉樹的前序遍歷 (20 分)

二叉樹的前序遍歷 (20 分)

給出一棵二叉樹,返回該二叉樹結點值的前序序列。你能用非遞迴方式完成麼?

函式介面定義:

vector<char> preOrder_iter(BTNode* b); //前序遍歷

其中 b 為指向二叉樹根結點的指標。
裁判測試程式樣例:

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

const int MaxSize = 500;

typedef struct TreeNode {
	char data;
	struct TreeNode *left, *right;
} BTNode;

void CreateTree(BTNode* &b, char str[]);//建立二叉樹,細節不表
vector<char> preOrder_iter(BTNode* b);//前序遍歷

int main() {
	char str[MaxSize];
	string ss;

	getline(cin, ss);
	strcpy(str, ss.c_str());

	BTNode* root = NULL;

	CreateTree(root, str);

	vector<char> v = preOrder_iter(root);
	for (int i = 0; i < v.size(); i++) {
		printf("%c", v[i]);
		if (i < v.size() - 1)
			printf(" ");
	}
	printf("\n");

	return 0;
}

/* 請在這裡填寫答案 */

輸入樣例1:(對應二叉樹如下)
在這裡插入圖片描述
輸出樣例1:

A B D C

輸入樣例2:(對應二叉樹如下)
在這裡插入圖片描述
輸出樣例2:

A B D C E

話不多說,上程式碼:

vector<char> preOrder_iter(BTNode* b)
{
    stack<BTNode*> ST;
    vector<char> VE;
    if(b==NULL)return VE;
    else
        ST.push(b);
    while(!ST.empty())
    {
        BTNode* temp=ST.top();
        ST.pop();
        VE.push_back(temp->data);
        if(temp->right!=NULL)
        {
            ST.push(temp->right);
        }
        if(temp->left!=NULL)
        {
            ST.push(temp->left);
        }
    }
    return VE;
}