C++의 같은 트리

두 개의 이진 트리가 있다고 가정합니다. 같은지 아닌지 확인하는 함수를 정의해야 합니다. 우리는 바이너리 트리가 구조적으로 동일하고 노드가 동일한 값을 가질 때 동일한 것으로 간주된다는 것을 알고 있습니다.

따라서 입력이 [1,2,3],[1,2,3]과 같으면 출력은 True

가 됩니다.

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

isSameTree라는 함수를 정의하면 두 개의 트리 노드 p와 q가 필요합니다.
p가 NULL과 같고 q가 NULL과 같으면 -
- true를 반환
p가 NULL과 같거나 q가 NULL과 같으면 -
- 거짓을 반환
p의 val이 p의 val과 같고 isSameTree(p의 왼쪽, q의 왼쪽)가 참이고 isSameTree(p의 오른쪽, q의 오른쪽)가 참이면
- true를 반환
거짓을 반환

예시

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

#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 isSameTree(TreeNode *p, TreeNode* q){
      if (p == NULL && q == NULL)
         return true;
      if (p == NULL || q == NULL)
         return false;
      if (p->val == q->val && isSameTree(p->left, q->left) && isSameTree(p->right, q->right))
         return true;
      return false;
   }
};
main(){
   Solution ob;
   vector<int> v = {1,2,3}, v1 = {1,2,3};
   TreeNode *root1 = make_tree(v);
   TreeNode *root2 = make_tree(v1);
   cout << (ob.isSameTree(root1, root2));
}

입력

{1,2,3}, {1,2,3}