LeetCode算法题-N-ary Tree Preorder Traversal（Java实现）

佚名 5年前 (2019-03-06) 算法 830人围观抢沙发百度已收录

这是悦乐书的第268次更新，第282篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第135题（顺位题号是589）。给定一个n-ary树，返回其节点值的前序遍历。例如，给定一个3-ary树：

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

其前序遍历结果为：[1,3,5,6,2,4]。

本次解题使用的开发工具是eclipse，jdk使用的版本是1.8，环境是win7 64位系统，使用Java语言编写和测试。

02 第一种解法

利用递归。二叉树前序遍历是从根节点开始，然后是左子树，最后是右子树，题目给的是N-ary树，但是遍历节点的顺序还是和二叉树一样，从根节点开始，然后对其子节点（list数组）进行遍历，其子节点进行前序遍历，和题目性质一样，可以直接使用递归的方式参与计算。

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

    public Node() {}

    public Node(int _val,List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
    public List<Integer> preorder(Node root) {
        if (root == null) {
            return new ArrayList<Integer>();
        }
        List<Integer> list = new ArrayList<Integer>();
        getValue(root, list);
        return list;
    }
    public List<Integer> getValue(Node root, List<Integer> list) {
        if (root == null) {
            return null;    
        } 
        list.add(root.val);
        if (root.children != null) {
            for (Node n : root.children) {
                getValue(n, list);
            }    
        }
        return list;
    }
}

03 第二种解法

我们也可以使用迭代的方式来实现前序遍历。这里我们使用栈，借助其先进后出的特性，有一点需要注意的是，在处理其子节点时，我们需要从后往前入栈，这样出栈的时候就是从前往后了。

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

    public Node() {}

    public Node(int _val,List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
    public List<Integer> preorder(Node root) {
        if (root == null) {
            return new ArrayList<Integer>();
        }
        List<Integer> list = new ArrayList<Integer>();
        Stack<Node> stack = new Stack<Node>();
        stack.push(root);
        while (!stack.isEmpty()) {
            Node temp = stack.pop();
            list.add(temp.val);
            if (temp.children != null) {
                for (int i=temp.children.size()-1; i >= 0; i--) {
                    stack.push(temp.children.get(i));
                }
            }
        }
        return list;
    }
}