ลบโหนดใบที่มีค่า x ใน Binary Tree ใน C++

บทนำ

โจทย์เดียวกับบทความก่อนหน้า (ID 1004078489) แต่นามพารามิเตอร์ต่างกัน (x แทน 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 x) {
    if (root == NULL) {
        return nullptr;
    }
    root->left = deleteLeafNodes(root->left, x);
    root->right = deleteLeafNodes(root->right, x);
    if (root->data == x && root->left == NULL && root->right == NULL) {
        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

หมายเหตุ

บทความนี้ซ้ำกับ ID 1004078489 เพียงแค่เปลี่ยนชื่อตัวแปร k → x เนื้อหาเหมือนกันทุกอย่าง

สรุป

Post-order Recursive deletion of leaf nodes with specific value รูปแบบโค้ดมาตรฐาน