LintCode: Binary Tree Preorder Traversal

C++，递归

/**
 * 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 binary tree.
     * @return: Preorder in vector which contains node values.
     */
    vector<int> preorderTraversal(TreeNode *root) {
        // write your code here
        vector<int> result;
        if (root == NULL) {
            return result;
        }
        result.insert(result.end(), root->val);
        if (root->left != NULL) {
            vector<int> left = preorderTraversal(root->left);
            result.reserve(result.size()+left.size());
            result.insert(result.end(), left.begin(), left.end());
        }
        if (root->right != NULL) {
            vector<int> right = preorderTraversal(root->right);
            result.reserve(result.size()+right.size());
            result.insert(result.end(), right.begin(), right.end());
        }
        return result;
    }
};

C++，递归，辅助函数

/**
 * 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 binary tree.
     * @return: Preorder in vector which contains node values.
     */
    vector<int> preorderTraversal(TreeNode *root) {
        // write your code here
        vector<int> result;
        if (root == NULL) {
            return result;
        } else {
            preorderCore(root, result);
        }
        return result;
    }
    void preorderCore(TreeNode *root, vector<int> &result) {
        if (root == NULL) {
            return ;
        }
        result.insert(result.end(), root->val);
        if (root->left != NULL) {
            preorderCore(root->left, result);
        }
        if (root->right != NULL) {
            preorderCore(root->right, result);
        }
    }
};

 C++，非递归

/**
 * 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 binary tree.
     * @return: Preorder in vector which contains node values.
     */
    vector<int> preorderTraversal(TreeNode *root) {
        // write your code here
        vector<int> result;
        if (root == NULL) {
            return result;
        }
        stack<TreeNode *> sta;
        TreeNode *cur = root;
        while (cur!=NULL || !sta.empty()) {
            while (cur != NULL) {
                result.insert(result.end(), cur->val);
                sta.push(cur);
                cur = cur->left;
            }
            cur = sta.top();
            sta.pop();
            cur = cur->right;
        }
        return result;
    }
};