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
结论
如果您在本教程中遇到任何问题,请在评论区提出。
广告