使用 C++ 在沒有遞迴的情況下列印從根到葉的路徑的程式


在本教程中,我們將討論一個程式,該程式列印從根節點到給定二叉樹中所有葉節點的路徑。

例如,假設我們有以下二叉樹

在這個二叉樹中,我們有 4 個葉節點。因此,我們可以從根節點到葉節點有 4 條路徑。

要解決這個問題,我們將使用迭代方法。在對二叉樹執行前序遍歷時,我們可以將父指標儲存在對映中。每當在遍歷過程中遇到葉節點時,我們就可以使用父指標輕鬆列印其從根節點的路徑。

示例

#include <bits/stdc++.h>>
using namespace std;
struct Node{
   int data;
   struct Node *left, *right;
};
//to create a new node
Node* create_node(int data){
   Node* node = new Node;
   node->data = data;
   node->left = node->right = NULL;
   return node;
}
//printing the path from root to leaf
void print_cpath(Node* curr, map<Node*, Node*> parent){
   stack<Node*> nodes_stack;
   while (curr){
      nodes_stack.push(curr);
      curr = parent[curr];
   }
   while (!nodes_stack.empty()){
      curr = nodes_stack.top();
      nodes_stack.pop();
      cout << curr->data << " ";
   }
cout << endl;
}
//to perform pre order traversal
void preorder_traversal(Node* root){
   if (root == NULL)
      return;
   stack<Node*> nodeStack;
   nodeStack.push(root);
   map<Node*, Node*> parent;
   parent[root] = NULL;
   while (!nodeStack.empty()){
      Node* current = nodeStack.top();
      nodeStack.pop();
      if (!(current->left) && !(current->right))
         print_cpath(current, parent);
      if (current->right){
         parent[current->right] = current;
         nodeStack.push(current->right);
      }
      if (current->left){
         parent[current->left] = current;
         nodeStack.push(current->left);
      }
   }
}
int main(){
   Node* root = create_node(101);
   root->left = create_node(82);
   root->right = create_node(23);
   root->left->left = create_node(34);
   root->left->right = create_node(55);
   root->right->left = create_node(29);
   preorder_traversal(root);
   return 0;
}

輸出

101 82 34
101 82 55
101 23 29

更新時間: 2019 年 11 月 1 日

128 次瀏覽

開啟你的職業生涯

完成課程,獲得認證

開始吧
廣告
© . All rights reserved.