参考文献链接:代码随想录
本人代码是Java版本的,如有别的版本需要请上代码随想录网站查看。
226.翻转二叉树
解题思路
翻转二叉树的思路还是比较好想的,其实就是把每个节点的左右子树交换即可。
可以用递归法和迭代法实现,加深对两种方法的理解。
代码示例
//递归法
class Solution {
public TreeNode invertTree(TreeNode root) {
reverseTree(root);
return root;
}
public void reverseTree(TreeNode node){
if(node == null) return;
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
reverseTree(node.left);
reverseTree(node.right);
}
}
//迭代法
class Solution {
public TreeNode invertTree(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
int len = queue.size();
while(len-- > 0){
TreeNode node = queue.poll();
if(node != null){
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
queue.offer(node.left);
queue.offer(node.right);
}
}
}
return root;
}
}
101. 对称二叉树
解题思路
该题目要判断二叉树是否对称,其实跟翻转区别不大,翻转是左右交换,对称只需要判断左右是否一样即可。
代码示例
//递归
class Solution {
public boolean isSymmetric(TreeNode root) {
return symmetric(root.left,root.right);
}
public boolean symmetric(TreeNode left,TreeNode right){
if(left== null && right == null){
return true;
}
if(left == null || right == null || left.val != right.val){
return false;
}
return symmetric(left.left,right.right) && symmetric(left.right,right.left);
}
}
//迭代
class Solution {
public boolean isSymmetric(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root.left);
queue.offer(root.right);
while(!queue.isEmpty()){
TreeNode left = queue.poll();
TreeNode right = queue.poll();
if(left==null&&right==null){
continue;
}
if(left==null||right==null||left.val!=right.val){
return false;
}
queue.offer(left.left);
queue.offer(right.right);
queue.offer(left.right);
queue.offer(right.left);
}
return true;
}
}
104.二叉树的最大深度
解题思路
其实这几道题都是跟层序遍历有关系,只要搞明白昨天的层序遍历即可。这道题只需要层序遍历然后计数有几层即可。
代码示例
class Solution {
public int maxDepth(TreeNode root) {
if (root == null) return 0;
Queue<TreeNode> que = new LinkedList<>();
que.offer(root);
int depth = 0;
while (!que.isEmpty())
{
int len = que.size();
while (len > 0)
{
TreeNode node = que.poll();
if (node.left != null) que.offer(node.left);
if (node.right != null) que.offer(node.right);
len--;
}
depth++;
}
return depth;
}
}
111.二叉树的最小深度
解题思路
这道题思路跟求最大深度一样,只不过要明白最小深度是什么?
最小深度是从根节点到最近叶子节点的最短路径上的节点数量
叶子节点就是左右孩子为null,所以我们加一个判断条件即可。
代码示例
class Solution {
public int minDepth(TreeNode root) {
if (root == null) return 0;
Queue<TreeNode> que = new LinkedList<>();
que.offer(root);
int depth = 0;
while (!que.isEmpty())
{
int len = que.size();
while (len > 0)
{
TreeNode node = que.poll();
//当出现叶子节点就return
if(node.left == null && node.right == null){
return ++depth;
}
if (node.left != null) que.offer(node.left);
if (node.right != null) que.offer(node.right);
len--;
}
depth++;
}
return depth;
}
}