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

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类
     */
    private static ArrayList<Integer> arrayList = new ArrayList<>();
    public TreeNode flattenTree (TreeNode root) {
        if (root == null) {
            return null;
        }
        search(root);
        TreeNode result = new TreeNode(-1);
        TreeNode current = result;
        for (int i : arrayList) {
            current.right = new TreeNode(i);
            current = current.right;
        }
        return result.right;
    }
    public void search(TreeNode root) {
        if (root == null) {
            return;
        }
        arrayList.add(root.val);
        search(root.left);
        search(root.right);
    }

}

本题知识点分析：

1.二叉树前序遍历

2.深度优先搜索

3.有序集合存取

4.节点赋值和遍历

本题解题思路分析：

1.先利用先序遍历获取所有的节点

2.取出集合中的节点，并在新创建的根节点中进行连接，只连接右节点

3.最后返回虚拟头结点.next

本题使用编程语言: Java