LeetCode.589. N 叉树的前序遍历

最新推荐文章于 2024-11-14 17:21:00 发布

原创最新推荐文章于 2024-11-14 17:21:00 发布 · 512 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #java #leetcode #数据结构

leetcode刷题专栏收录该内容

41 篇文章

订阅专栏

本文介绍了如何将二叉树的前序遍历策略扩展到N叉树，指出两者思路相似，只需将二叉树的根左右概念替换为根孩子。通过递归函数展示了N叉树前序遍历的代码实现。

题目

589. N 叉树的前序遍历

分析

我们之前有做过LeetCode的 144. 二叉树的前序遍历，其实对于 N 插树来说和二叉树的思路是一模一样的。
二叉树的前序遍历是【根左右】
N叉树的前序遍历顺序是【根孩子】，你可以把二叉树的【根左右】想象成【根孩子】，因为左右就是孩子。

代码

先看一下二叉树前序遍历的代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        func(root,res);
        return res;
    }

    void func(TreeNode cur,List<Integer> res) {
        if(cur == null) return;
        // 先记录根节点
        res.add(cur.val);
        // 遍历左子树
        func(cur.left,res);
        // 遍历右子树
        func(cur.right,res);
    }
}

只需要改动一点就是N叉树的前序遍历代码，如下：

/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
    public List<Integer> preorder(Node root) {
        List<Integer> res = new ArrayList<>();
        if(root == null) return res;
        pre(root,res);
        return res;
    }

    void pre(Node cur,List<Integer> res) {
        if(cur == null) return ;
        // 先记录当前节点
        res.add(cur.val);
        // 再遍历孩子
        for(Node node : cur.children) {
            pre(node,res);
        }
    }
}