C++에서 등가 이진 트리 뒤집기

이진 트리가 있다고 가정합니다. 이진 트리를 뒤집어야 합니다. 반전은 다음을 나타냅니다. 임의의 노드를 선택하고 왼쪽 및 오른쪽 하위 하위 트리를 교환합니다. 이제 이진 트리 X는 몇 번의 뒤집기 작업 후에 X에서 Y를 만들 수 있는 경우에만 이진 트리 Y와 같은 뒤집기입니다. 두 개의 이진 트리가 대칭 이동인지 여부를 결정하는 메서드를 작성해야 합니다. 트리는 루트 노드 root1 및 root2에 의해 제공됩니다. 따라서 나무가 -

값이 1, 3, 5인 노드를 뒤집으면 출력은 true가 됩니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

• 하나의 재귀 함수 solve()를 정의하면 두 개의 트리 t1과 t2가 필요합니다.

• root1과 root2가 null이면 true를 반환합니다.

• 그렇지 않으면 root1이 null이거나 root2가 null이면 false를 반환합니다.

• 그렇지 않으면 (t1 및 t2 둘 다 왼쪽 하위 트리가 없음) 또는 (t1 및 t2 둘 다 왼쪽 하위 트리가 있고 이 두 노드의 왼쪽 하위 트리 값이 동일함)

  • 해결(루트1의 왼쪽, 루트2의 왼쪽) 및 해결(루트1의 오른쪽, 루트2의 오른쪽)

• 그렇지 않으면

  • 해결(루트1의 왼쪽, 루트2의 오른쪽) 및 해결(루트1의 오른쪽, 루트2의 왼쪽)

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

예시

#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 insert(TreeNode **root, int val){
   queue<TreeNode*> q;
   q.push(*root);
   while(q.size()){
      TreeNode *temp = q.front();
      q.pop();
      if(!temp->left){
         if(val != NULL)
            temp->left = new TreeNode(val);
         else
            temp->left = new TreeNode(0);
         return;
      }else{
         q.push(temp->left);
      }
      if(!temp->right){
         if(val != NULL)
            temp->right = new TreeNode(val);
         else
            temp->right = new TreeNode(0);
         return;
      }else{
         q.push(temp->right);
      }
   }
}
TreeNode *make_tree(vector<int> v){
   TreeNode *root = new TreeNode(v[0]);
   for(int i = 1; i<v.size(); i++){
      insert(&root, v[i]);
   }
   return root;
}
class Solution {
   public:
   bool flipEquiv(TreeNode* root1, TreeNode* root2) {
      if(!root1 && !root2)return true;
      else if(!root1 || !root2)return false;
      else if(root1->val != root2->val) return false;
      else if((!root1->left && !root2->left) || (root1->left && root2->left && root1->left->val ==          root2-      >left->val)){
         return flipEquiv(root1->left, root2->left) && flipEquiv(root1->right, root2->right);
      }else{
         return flipEquiv(root1->left, root2->right) && flipEquiv(root1->right, root2->left);
      }
   }
};
main(){
   vector<int> v = {1,2,3,4,5,6,NULL,NULL,NULL,7,8};
   TreeNode *r1 = make_tree(v);
   vector<int> v1 = {1,3,2,NULL,6,4,5,NULL,NULL,NULL,NULL,NULL,NULL,8,7};
   TreeNode *r2 = make_tree(v);
   Solution ob;
   cout << (ob.flipEquiv(r1, r2));
}

입력

[1,2,3,4,5,6,null,null,null,7,8]
[1,3,2,null,6,4,5,null,null,null,null,8,7]

출력

1