题解 | #牛群的树形结构展开#

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 flattenTree (TreeNode root) {
        // write code here
        if (root == null) return null;
        List<TreeNode> list = new ArrayList<>();
        preOrder(root, list);
        for (int i = 0; i < list.size() - 1; i++) {
            TreeNode node = list.get(i);
            node.left = null;
            node.right = list.get(i + 1);
        }
        return list.get(0);
    }
    private void preOrder(TreeNode root, List<TreeNode> list) {
        if (root == null) return;
        list.add(root);
        preOrder(root.left, list);
        preOrder(root.right, list);
    }
}

• 根据题意，可以基于二叉树的前序遍历进行求解
• 首先，边界值处理：如果 root 为null，则直接返回null
• 定义一个List集合，用于记录二叉树前序遍历的结果
• 前序遍历后，对 list 进行遍历 0 到 size - 1 次，每次获取一个节点
• 将节点的左子树赋值为null
• 将节点的右子树赋值为下一个 list 的元素
• 最后返回 list 的第一个元素即可