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 次浏览

开启你的职业生涯

完成课程获得认证

开始学习
广告