爲什麼寫這篇帖子
這篇帖子的代碼是在上大學修數據結構的時候寫的,寫完就上傳提供下載了,那會比較貪心,設置的下載積分比較高。工作後慢慢習慣分享,偶然看到當時上傳的簡單幾行代碼還要積分。乾脆直接把代碼拿過來寫成帖子也省去了路過同仁下載的麻煩。
這篇帖子都寫了什麼
如題,這篇帖子給出了二叉樹三種遍歷算法的非遞歸實現。目前,我只給出了算法的具體代碼實現,具體的算法描述並沒有提及(貌似也沒有必要說,算法很簡單)。
順便提一下本文中的算法使用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;
}
結束語
以上就是二叉樹三種遍歷算法的遞歸與非遞歸實現的具體代碼,如有不明白的地方歡迎留言交流,若帖子中有錯誤的地方同樣歡迎留言批評指正,在此謝過路過的各位大神。