Advertisement
jayati

Delete Node in a BST

May 15th, 2024
532
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.78 KB | None | 0 0
  1. /**
  2.  * Definition for a binary tree node.
  3.  * struct TreeNode {
  4.  *     int val;
  5.  *     TreeNode *left;
  6.  *     TreeNode *right;
  7.  *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
  8.  *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
  9.  *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  10.  * };
  11.  */
  12. class Solution {
  13. public:
  14.  
  15. TreeNode* maxValue(TreeNode*root){
  16.     if(root==NULL){
  17.         return 0;
  18.     }
  19.     TreeNode*temp =root;
  20.     while(temp->right!=NULL){
  21.         temp= temp->right;
  22.     }
  23.     return temp;
  24. }
  25.     TreeNode* deleteNode(TreeNode* root, int key) {
  26.         if(root==NULL)
  27.             return root;
  28.        
  29.         if(root->val ==key){
  30.             // for 0 child
  31.             if(root->left==NULL && root->right ==NULL){
  32.                 delete root;
  33.                 return NULL;
  34.             }
  35.             // for 1 child
  36.             if(root->left!=NULL && root->right ==NULL){
  37.                 TreeNode*temp = root->left;
  38.                 delete root;
  39.                 return temp;
  40.             }
  41.              if(root->left==NULL && root->right !=NULL){
  42.                 TreeNode*temp = root->right;
  43.                 delete root;
  44.                 return temp;
  45.             }
  46.             // for 2 child
  47.             if(root->left!=NULL && root->right!=NULL){
  48.             int maxi = maxValue(root->left)->val;
  49.               root->val = maxi;
  50.               root->left =deleteNode(root->left,maxi);
  51.               return root;
  52.             }
  53.         }
  54.         else if(root->val >key){
  55.             root->left= deleteNode(root->left,key);
  56.             return root;
  57.         }
  58.         else{
  59.             root->right = deleteNode(root->right,key);
  60.             return root;
  61.         }
  62.         return root;
  63.     }
  64. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement