题解 | #翻转牛群结构# java

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return TreeNode类
     */
    public TreeNode invertTree (TreeNode root) {
        // write code here
         if (root == null) {
            return null;
        }

        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);

        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;

            if (node.left != null) {
                stack.push(node.left);
            }
            if (node.right != null) {
                stack.push(node.right);
            }
        }

        return root;
    }
}

这道题目考察的是二叉树的翻转。下面是对代码的解释大纲：

1. 首先检查根节点是否为null，如果是null，则直接返回null。
2. 创建一个栈用于存储待处理的节点，将根节点入栈。
3. 进入循环，只要栈不为空，就执行以下步骤：弹出栈顶节点，保存在变量node中。交换node节点的左右子节点，使用一个临时变量temp进行交换。如果node的左子节点不为null，则将左子节点入栈。如果node的右子节点不为null，则将右子节点入栈。
4. 循环结束后，返回翻转后的二叉树的根节点。

该解法使用了迭代的方式，通过栈来辅助处理每个节点。从根节点开始，依次交换每个节点的左右子节点，并将非空子节点入栈，确保下一轮迭代可以处理它们的子节点。