[binary tree] 993 Cousins in Binary Tree

Given the root of a binary tree with unique values and the values of two different nodes of the tree x and y, return true if the nodes corresponding to the values x and y in the tree are cousins, or false otherwise.

Two nodes of a binary tree are cousins if they have the same depth with different parents.

Note that in a binary tree, the root node is at the depth 0, and children of each depth k node are at the depth k + 1.

The question ： Define cousin node ： Same number of layers , But the parent nodes are different .

Ideas ：dfs, Record the number of layers and the value of the parent node .

Returns an array , It contains the number of layers and the value of the parent node

class Solution {
    public boolean isCousins(TreeNode root, int x, int y) {
        int[] a = dfs(root,x,0,-1);
        int[] b = dfs(root,y,0,-1);
        if(a[0] == b[0]){
            return a[1] != b[1];
        }
        return false;
    }
    int[] dfs(TreeNode root,int x,int level,int p){
        if(root == null){
            return new int[]{};
        }
        if(root.val == x){
            return new int[]{level,p};
        }
        int[]a = dfs(root.left,x,level+1,root.val);
        int[]b = dfs(root.right,x,level+1,root.val);
        if(a.length == 0) return b;
        return a;
    }
}