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

牛群的树形结构展开

https://www.nowcoder.com/practice/07caea5438394f58afbe72cbe2eb2189?tpId=354&tqId=10591559&ru=/exam/oj&qru=/ta/interview-202-top/question-ranking&sourceUrl=%2Fexam%2Foj%3Fpage%3D1%26tab%3D%25E7%25AE%2597%25E6%25B3%2595%25E7%25AF%2587%26topicId%3D354

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
        List<TreeNode> list = new ArrayList<TreeNode>();
        dfs(root, list);
        int size = list.size();
        for (int i = 1; i < size; i++) {
            TreeNode pre = list.get(i - 1), curr = list.get(i);
            pre.left = null;
            pre.right = curr;
        }
        return root;
    }


    //先序遍历二叉树
    private void dfs(TreeNode root, List<TreeNode> list) { 
        if (root != null) {
            list.add(root); //存储节点
            dfs(root.left, list);
            dfs(root.right, list);
        }
    }
}

知识点分析:

  1. 二叉树:题目中牵涉到了二叉树的遍历和展开操作。
  2. 二叉树的遍历:这个问题要求将二叉树按照先序遍历的顺序展开,可以使用递归或迭代方法实现。
  3. 单链表的构建:在展开二叉树的同时,需要构建单链表结构,即每个节点的右子指针指向下一个节点。

题目解答方法分析:

  1. 定义一个树节点的类,表示二叉树的节点。
  2. 使用递归进行先序遍历展开:从根节点出发,递归地遍历左子树、右子树,将左子树置为空,右子树指向下一个节点。
  3. 在递归过程中,维护一个指向当前链表尾部的指针,以便将展开的节点依次连接起来。
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务