[LeetCode] 144. Binary Tree Preorder Traversal

题目描述

分析

二叉树前序遍历，非常简单，不过题目要求用递推法，不可用递归，不经常写的最好再熟悉下，虽然不难，面试中说不定会被问到。递推法只需额外开辟一个栈即可，每遍历完当前节点，按顺序将右孩子（如果有的话）和左孩子（如果有的话）依次入栈，下次遍历则直接从栈顶取就好，直到栈为空。

code

class Solution {
public:

	vector<int> preorderTraversal(TreeNode* root) {
		vector<int> res; //保存结果
		stack<TreeNode*> s;
		// 初始，将根节点入栈
		if (root != NULL) {
			s.push(root);
		}
		// 每次从栈顶取元素遍历，直至为空
		while (!s.empty()) {
			TreeNode* cur = s.top();
			s.pop();
			res.push_back(cur->val);
			if (cur->right != NULL) {	//先将右孩子入栈
				s.push(cur->right);
			}
			if (cur->left != NULL) {	//再将左孩子入栈
				s.push(cur->left);
			}
		}
		return res;
	}
};

话说这LeetCode的运行时间也不是每次都一样，不过大致差不多。