树的很多题目都是对根结点(或者当前结点) 进行处理之后, 再对子结点进行处理. 本题也是如此, 对根结点判断后, 再对其根节点的子结点进行判断处理. 同样也是有递归和非递归两种方式来解决.
递归的方法 :
class Solution {
public boolean isSymmetric(TreeNode root) {
if (root == null)
return true;
return isSymmetricHelper(root.left, root.right);
}
public boolean isSymmetricHelper(TreeNode left, TreeNode right) {
if (left == null && right == null)
return true;
if (left == null || right == null)
return false;
if (left.val != right .val)
return false;
return isSymmetricHelper(left.left, right.right) && isSymmetricHelper(left.right, right.left);
}
}
这里要注意的是左结点的右子树和右结点的左子树, 这个对应关系不能弄错.
非递归的方法 :
class Solution {
public boolean isSymmetric(TreeNode root) {
if (root == null)
return true;
Stack<TreeNode> nodes = new Stack<>();
TreeNode left, right;
if (root.left != null) {
if (root.right == null) return false;
nodes.push(root.left);
nodes.push(root.right);
} else if (root.right != null) {
return false;
}
while(!nodes.isEmpty()) {
if (nodes.size()%2 != 0) return false;
right = nodes.pop();
left = nodes.pop();
if (right.val != left.val)
return false;
if (left.left != null) {
if (right.right == null) return false;
nodes.push(left.left);
nodes.push(right.right);
} else if (right.right != null) {
return false;
}
if (right.left != null) {
if (left.right == null) return false;
nodes.push(left.right);
nodes.push(right.left);
} else if (left.right != null) {
return false;
}
}
return true;
}
}
这里需要注意的是, 无论是 left.left, right.right 还是 left.right, right.left, 在进栈的时候, 先处理左子树还是右子树, 前后要保持一致.