Binary Tree Paths二叉树路径

问题描述：

Given a binary tree, return all root-to-leaf paths.给定一个二叉树，返回所有从根节点到叶子节点的路径。

思路：

深度优先遍历中的前序遍历（给定子树，先访问根节点，再访问左节点，最后访问右节点），非递归的实现方式通过堆栈辅助。由于题目要求返回所有“根节点到叶子节点的路径”，因此使用两个堆栈，一个存放节点，另一个存放节点的路径（string形式）。

示例：

输入：{1,2,3,#,5}，输出:["1->2->5","1->3"]。输入：{}，输出：[]。

代码如下：

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param root: the root of the binary tree
     * @return: all root-to-leaf paths
     */
    vector<string> binaryTreePaths(TreeNode * root) {
        // write your code here
        stack<string> sr;
        stack<TreeNode*> stn;
        vector<string> r;  //result: path list
        if (root == NULL) {
            return r;
        } else if (root->left==NULL && root->right==NULL) {
            //leaf node
            r.push_back(to_string(root->val));
            return r;
        } 
        sr.push(to_string(root->val));
        stn.push(root);
        TreeNode *p;
        string sp;
        while (!stn.empty()) {
            // PreOrder Depth-First traverse
            p = stn.top();
            stn.pop();
            sp = sr.top();
            sr.pop();
            if (p->right != NULL) {
                if (p->right->left==NULL && p->right->right==NULL) {
                    //leaf node
                    r.push_back(sp+"->"+to_string(p->right->val));
                } else {
                    stn.push(p->right);
                    sr.push(sp+"->"+to_string(p->right->val));
                }
            }
            if (p->left != NULL) {
                if (p->left->left==NULL && p->left->right==NULL) {
                    //leaf node
                    r.push_back(sp+"->"+to_string(p->left->val));
                } else {
                    stn.push(p->left);
                    sr.push(sp+"->"+to_string(p->left->val));
                }
            }
        }
        return r;
    }
};