树-前序遍历之迭代写法

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

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 Solution1{

    public List<Integer> preorderTraversal(TreeNode root){

        List<Integer> list = new ArrayList<>();
        if(root == null)return list;
        Deque<TreeNode> stack = new LinkedList<>();
        while(!stack.isEmpty() || root != null){
            // 注意这里是双重循环，而不是单循环
            while(root != null){
                stack.push(root);
                list.add(root.val);
                root = root.left;
            }
            root = stack.pop();
            root = root.right;
        }
        return list;
    }

}

/*
发现：
程序非常喜欢使用多态写法
栈中存放的都是目前遍历到的根节点，入栈的时候他们是根节点，出栈的时候用到他们的右子树节点作为最后的使命；
循环中的四种情况需要自己一一体会什么时候才满足各式各样的条件：
1、栈空且节点空 --> 结束时
2、栈空且节点不空 --> 第一次遍历时、该节点的前面部分全部遍历完成
3、栈不空且节点空 --> 遇到空节点
4、栈不空且节点不空 --> 该节点是根节点的左子树
只有自己了解透所有可能出现的情况，才能用迭代；

*/

class test {
    public static void main(String[] args) {
        TreeNode node1 = new TreeNode(1);
        TreeNode node2 = new TreeNode(2);
        TreeNode node3 = new TreeNode(3);
        TreeNode node4 = new TreeNode(4);
        TreeNode node5 = new TreeNode(5);
        TreeNode node6 = new TreeNode(6);
        TreeNode node7 = new TreeNode(7);
        node1.left = node2;node1.right = node3;
        node2.left = node4;node2.right = node5;
        node3.left = node6;node3.right = node7;
        // 这里要用 List 接收，如果使用 ArrayList 接收则需要强制转型
//        List<Integer> res = new Solution2().preorderTraversal(node1);

        List<Integer> res = new Solution1().preorderTraversal(node1);
        for (int i = 0; i < res.size(); i++) {
            // 这里需要用到 get 函数
            System.out.println(res.get(i));
        }
    }
}