J 判断二叉树每个结点的权值是否关于根节点完全对称

如果二叉树每个结点的权值关于根节点完全对称 就输出Yes

Sample Input

2
7 //结点
1 2 3 //结点1的左孩子是结点2 右孩子是结点3
2 4 5
3 6 7
4 0 0
5 0 0
6 0 0
7 0 0
1 2 2 3 4 4 3 //权值
5
1 2 3
2 0 4
3 0 5
4 0 0
5 0 0
1 2 2 3 3

Sample Output

Yes
No

 # include <cstdio>
 # include <cstring>
 # define LL long long
 using namespace std ;

 bool flag  ;

 struct tree
 {
     int lson ;
     int rson ;
     int num ;
     int id ;
 }t[];

 void dfs(int r1 , int r2)
 {
     if (r1 ==  && r2 == )
         return ;
     if (r1 ==  && r2 !=  || r1 !=  && r2 ==  || t[r1].num != t[r2].num)
     {
         flag =  ;
         return ;
     }
     if (!flag)
         return ;
     dfs(t[r1].lson , t[r2].rson) ;
     dfs(t[r1].rson , t[r2].lson) ;
 }

 int main ()
 {
    // freopen("in.txt","r",stdin) ;
     int T ;
     scanf("%d" , &T) ;
     while (T--)
     {
         int n ;
         scanf("%d" , &n) ;
         int i ;
         int a ,b ,c ;
         for (i =  ; i <= n ;i++)
         {
             scanf("%d %d %d" , &a , &b , &c) ;
             t[a].lson = b ;
             t[a].rson = c ;
         }
         for (i =  ; i <= n ;i++)
             scanf("%d" , &t[i].num) ;
         flag =  ;

         dfs(t[].lson , t[].rson) ;
         if (flag)
             printf("Yes\n") ;
         else
             printf("No\n") ;
     }

     return  ;
 }