Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++ 프로그램에서 값이 k인 리프 노드 삭제

<시간/>

이 튜토리얼에서는 주어진 값으로 트리에서 리프 노드를 삭제하는 방법을 배울 것입니다.

문제를 해결하는 단계를 살펴보겠습니다.

  • 이진 트리에 대한 구조체 노드를 작성하십시오.

  • 트리를 순회(inorder, preorder, postorder)하고 모든 데이터를 출력하는 함수를 작성하십시오.

  • 구조체로 노드를 생성하여 트리를 초기화합니다.

  • x 값을 초기화합니다.

  • 주어진 값으로 리프 노드를 삭제하는 함수를 작성하십시오. 두 개의 인수 루트 노드와 k 값을 허용합니다.

    • 루트가 null인 경우 반환합니다.

    • 삭제 후 루트의 왼쪽 노드를 새 루트로 교체합니다.

    • 루트의 오른쪽 노드와 동일합니다.

    • 현재 루트 노드 데이터가 k와 같고 리프 노드이면 null 포인터를 반환합니다.

    • 루트 노드 반환

예시

코드를 봅시다.

#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

결론

튜토리얼에서 질문이 있는 경우 댓글 섹션에 언급하세요.