题:
思:
用第二个队列来保存在同一层的所有结点,
如果能找到x和y,且它们不拥有同一个父节点,则返回true
如果只能找到一个x或者y,则返回false,因为此时这意味着另一个x或者y,和我们这一层找到的x或者y一定不在同一层
如果既没找到x又没找到y,则再找下一层
最后也没找到,返回false
码:
public boolean isCousins(TreeNode root, int x, int y) {
if (root.val == x || root.val == y)
return false;
// 如果深度相同,且不是同一个父亲,则返回true
// 当深度为h的时候,应该弹出2的h次方个结点
Queue<TreeNode> queue = new LinkedList<TreeNode>();
Queue<TreeNode> tempQueue = new LinkedList<TreeNode>();
queue.offer(root);
int count = 0;
while (!queue.isEmpty()) {
// 弹出所有的元素,此时弹出的元素都在同一层
while (!queue.isEmpty()) {
TreeNode poll = queue.poll();
tempQueue.offer(poll);
}
// 分别将这些元素的孩子加入队列queue,如果两个孩子正好是x和y,返回false
// 如果所有的孩子加完以后,只有一个x或者y,则返回false
// 如果能找到不同父亲的两个孩子,则返回true
while (!tempQueue.isEmpty()) {
TreeNode poll = tempQueue.poll();
if (poll.left != null && poll.right != null) {
if ((poll.left.val == x && poll.right.val == y) || (poll.left.val == y && poll.right.val == x))
return false;
queue.offer(poll.left);
if (poll.left.val == x || poll.left.val == y) count++;
queue.offer(poll.right);
if (poll.right.val == x || poll.right.val == y) count++;
} else if (poll.left != null){
queue.offer(poll.left);
if (poll.left.val == x || poll.left.val == y) count++;
}else if(poll.right != null){
queue.offer(poll.right);
if (poll.right.val == x || poll.right.val == y) count++;
}
}
if(count == 1)return false;
if(count == 2)return true;
}
return false;
}