Giả sử chúng ta có một cây nhị phân, trong đó giá trị của mọi nút là 0 hoặc 1. Chúng ta phải tìm cùng một cây mà mọi cây con không chứa số 1 đã bị xóa. Vì vậy, nếu cây như -
Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -
-
Định nghĩa một phương thức đệ quy giải quyết (), điều này sẽ lấy nút. phương pháp sẽ như thế nào -
-
Nếu nút là null, thì trả về null
-
bên trái của nút:=giải quyết (bên trái của nút)
-
bên phải của nút:=giải quyết (bên phải của nút)
-
nếu bên trái của nút là null và bên phải của nút cũng là null và giá trị của nút là 0, thì trả về null
-
nút trả lại
Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -
Ví dụ
#include <bits/stdc++.h>
using namespace std;
class TreeNode{
public:
int val;
TreeNode *left, *right;
TreeNode(int data){
val = data;
left = NULL;
right = NULL;
}
};
void inorder(TreeNode *root){
if(root){
inorder(root->left);
cout << root->val << ", ";
inorder(root->right);
}
}
class Solution {
public:
TreeNode* pruneTree(TreeNode* node) {
if(!node)return NULL;
node->left = pruneTree(node->left);
node->right = pruneTree(node->right);
if(!node->left && !node->right && !node->val){
return NULL;
}
return node;
}
};
main(){
TreeNode *root = new TreeNode(1);
root->left = new TreeNode(1);
root->right = new TreeNode(0);
root->left->left = new TreeNode(1);
root->left->right = new TreeNode(1);
root->right->left = new TreeNode(0);
root->right->right = new TreeNode(1);
root->left->left->left = new TreeNode(0);
Solution ob;
inorder(ob.pruneTree(root));
} Đầu vào
TreeNode *root = new TreeNode(1); root−>left = new TreeNode(1); root−>right = new TreeNode(0); root−>left−>left = new TreeNode(1); root−>left−>right = new TreeNode(1); root−>right−>left = new TreeNode(0); root−>right−>right = new TreeNode(1); root−>left−>left−>left = new TreeNode(0);
Đầu ra
1, 1, 1, 1, 0, 1,