C++:二叉樹的非遞歸遍歷實現

爲什麼寫這篇帖子

這篇帖子的代碼是在上大學修數據結構的時候寫的,寫完就上傳提供下載了,那會比較貪心,設置的下載積分比較高。工作後慢慢習慣分享,偶然看到當時上傳的簡單幾行代碼還要積分。乾脆直接把代碼拿過來寫成帖子也省去了路過同仁下載的麻煩。

這篇帖子都寫了什麼

如題,這篇帖子給出了二叉樹三種遍歷算法的非遞歸實現。目前,我只給出了算法的具體代碼實現,具體的算法描述並沒有提及(貌似也沒有必要說,算法很簡單)。

順便提一下本文中的算法使用C++描述。

具體聊聊代碼

二叉樹數據結構

//定義二叉樹結構
struct TreeNode
{
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

遍歷算法實現

非遞歸先序遍歷

//非遞歸先序遍歷
vector<int> preorderTraversal(TreeNode* root) {
	vector<int> res;

	stack<TreeNode*> st;
	st.push(root);
	while (!st.empty() && root != nullptr)
	{
		auto temp = st.top();
		st.pop();
		if (temp == nullptr)
		{
			continue;
		}
		else
		{
			res.push_back(temp->val);
			st.push(temp->right);
			st.push(temp->left);
		}
	}

	return res;
}

非遞歸中序遍歷

//非遞歸中序遍歷
vector<int> inorderTraversal(TreeNode* root) {
	vector<int> res;
	if (root == nullptr)
	{
		return res;
	}

	stack<TreeNode*> s;
	while (root != nullptr || !s.empty())
	{
		while (root)
		{
			s.push(root);
			root = root->left;
		}
		if (!s.empty())
		{
			root = s.top();
			res.push_back(root->val);
			s.pop();
			root = root->right;
		}
	}

	return res;
}

非遞歸後序遍歷

//非遞歸後序遍歷
vector<int> postorderTraversal(TreeNode* root) {
	vector<int> res;
	if (root == nullptr)
	{
		return res;
	}

	stack<TreeNode*> s;
	vector<TreeNode*> v;

	s.push(root);

	while (!s.empty())
	{
		while (s.top()->left&&find(v.begin(), v.end(), s.top()->left) == v.end())
		{
			s.push(s.top()->left);
		}

		root = s.top();
		if (root->right&&find(v.begin(), v.end(), s.top()->right) == v.end())//認爲沒有左節點
		{
			s.push(root->right);
		}
		else//認爲沒有右節點
		{
			res.push_back(root->val);
			v.push_back(root);
			s.pop();
		}
	}
	return res;
}

結束語

以上就是二叉樹三種遍歷算法的遞歸與非遞歸實現的具體代碼,如有不明白的地方歡迎留言交流,若帖子中有錯誤的地方同樣歡迎留言批評指正,在此謝過路過的各位大神。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章