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
結論
如果您在本教程中遇到任何問題,請在評論區提出。
廣告