LeetCode刷题笔记（Binary Tree Preorder Traversal）

今天赶早刷了一道题，感觉这道题还是挺重要的，下面就和大家分享一下经验吧！

题目如下：

Given a binary tree, return the preorder traversal of its nodes' values.

Example:

Input: [1,null,2,3]
   1
    \
     2
    /
   3

Output: [1,2,3]
Follow up: Recursive solution is trivial, could you do it iteratively?

题意分析： 

给定一个二分搜索树，返回前序遍历后节点的值。如果不使用递归方法，用其他的方法呢？

解答如下：

方法一（递归法）

树的遍历，一般有前序遍历，中序遍历，后序遍历和层序遍历，前三种遍历其实可以用一种方法来辨别，先将每个节点标注左中右三个蓝点，然后按照图中黑线的方式遍历整个整棵树，若是前序遍历则取出左蓝点对应的值，若是中序遍历则取出中蓝点对应的值，若是后序遍历则取出右蓝点对应的值，最后按取出的先后顺序对这些值进行存储或者打印，具体操作见下图：

由于是前序遍历，所以编写的递归代码如下：

class Solution{
public:
    vector<int> preorderTraversal( TreeNode* node ){
        vector<int> res;
        if( !node )
            return res;
        recursive(node, res);
        return res;
    }

    void recursive( TreeNode* node, vector<int>& res ){
        if( node )
        {
            res.push_back( node -> val );                 //此处操作便为前序遍历
            recursive( node -> left, res);
            recursive( node -> right, res);
        }
    }
};

 提交后的结果如下： 

 

方法二（非递归法）

使用栈去模拟系统栈，首先创建一个指令的结构体和一个用于存放指令的栈，然后弄清指令执行的先后顺序，根据栈“先进后出”的原则依次将指令压入栈中，最后返回所需要的结果。

//定义指令结构体
struct Instruction{
    string s;
    TreeNode* node;
    Instruction(string s, TreeNode* node): s(s), node(node){}
};

//模拟系统栈
class Solution{
public:
    vector<int> preorderTraversal( TreeNode* node ){

        vector<int> res;
        if ( !node ) return res;

        stack<Instruction> stack;
        stack.push( Instruction("Go", node) );         //压入第一个起始指令
        while ( !stack.empty() ){                      //若栈中有指令则仍需继续执行指令

            Instruction instruction = stack.top();     //取指令
            stack.pop();
            //执行指令
            if ( instruction.s == "Out" ) res.push_back( instruction.node->val ); 
            else{
                assert( instruction.s == "Go" );
                //根据先进后出的原则压入指令
                if ( instruction.node -> right )
                    stack.push( Instruction( "Go", instruction.node->right ));
                if ( instruction.node -> left )
                    stack.push( Instruction( "Go", instruction.node->left ));
                stack.push( Instruction( "Out", instruction.node ) );  //此处操作便为前序遍历
            }
        }
     return res;
    }
};

提交后的结果如下：

 

日积月累，与君共进，增增小结，未完待续。