C++程式中刪除值為k的葉子節點


在本教程中,我們將學習如何從樹中刪除具有給定值的葉子節點。

讓我們看看解決問題的步驟。

  • 為二叉樹編寫一個結構體Node。

  • 編寫一個函式來遍歷(中序、前序、後序)樹並列印所有資料。

  • 透過使用結構體建立節點來初始化樹。

  • 初始化x值。

  • 編寫一個函式來刪除具有給定值的葉子節點。它接受兩個引數:根節點和k值。

    • 如果根節點為空,則返回。

    • 刪除後,用新的根節點替換根節點的左節點。

    • 根節點的右節點也是如此。

    • 如果當前根節點資料等於k並且它是一個葉子節點,則返回空指標。

    • 返回根節點

示例

讓我們看看程式碼。

 線上演示

#include <bits/stdc++.h>
using namespace std;
struct Node {
   int data;
   struct Node *left, *right;
};
struct Node* newNode(int data) {
   struct Node* newNode = new Node;
   newNode->data = data;
   newNode->left = newNode->right = NULL;
   return newNode;
}
Node* deleteLeafNodes(Node* root, int k) {
   if (root == NULL) {
      return nullptr;
   }
   root->left = deleteLeafNodes(root->left, k);
   root->right = deleteLeafNodes(root->right, k);
   // checking the current node data with k
   if (root->data == k && root->left == NULL && root->right == NULL) {
      // deleting the node
      return nullptr;
   }
   return root;
}
void inorder(Node* root) {
   if (root == NULL) {
      return;
   }
   inorder(root->left);
   cout << root->data << " ";
   inorder(root->right);
}
int main(void) {
   struct Node* root = newNode(1);
   root->left = newNode(2);
   root->right = newNode(3);
   root->left->left = newNode(3);
   root->left->right = newNode(4);
   root->right->right = newNode(5);
   root->right->left = newNode(4);
   root->right->right->left = newNode(4);
   root->right->right->right = newNode(4);
   deleteLeafNodes(root, 4);
   cout << "Tree: ";
   inorder(root);
   cout << endl;
   return 0;
}

輸出

如果執行上述程式碼,則會得到以下結果。

Tree: 3 2 1 3 5

結論

如果您在本教程中遇到任何問題,請在評論區提出。

更新於:2021年1月27日

77 次瀏覽

開啟你的職業生涯

完成課程獲得認證

開始學習
廣告