迭代遍历二叉树

今天想着学动态规划，不过发现貌似有点难度，功力暂时还不够。然后看到了一个二叉树迭代遍历方法的题目，感觉挺有意思的，就写了个。

通过这个可以总结的写代码的经验。先上代码吧。

前序：

/*
前序遍历其实是比较简单的，因为递归的实现本来就是用的栈的原理，所以直接用栈来写二叉树的遍历就好。
前序遍历就是第一次遇到就把遇到的数据处理。所以就遇到就输出，一直到最左叶子节点。
然后开始处理右边的节点，右子树为空继续pop到不为空为止，之后对右子树进行之前进行的操作就行。
*/
vector<int> preorderTraversal(TreeNode *root) {
// write your code here
vector<int> result;
stack<TreeNode *> myStack;
TreeNode *current = root;
while(current!=NULL||!myStack.empty()){
while(current!=NULL){
result.push_back(current->val);
myStack.push(current);
current = current -> left;
}
while(!myStack.empty()){
current = myStack.top();
myStack.pop();
if(current->right!=NULL)
current = current->right;
else
current = NULL;
}
}
return result;
}

接下来是中序遍历：

/*
中序遍历和前序遍历一样，只是处理数据是在第二次，所以一直压栈，直到pop之后再处理数据就是第二次了。
*/
vector<int> inorderTraversal(TreeNode *root) {
// write your code here
vector<int> result;
stack<TreeNode *> myStack;
TreeNode *current = root;
while(!myStack.empty()||current!=NULL){
while(current!=NULL){
myStack.push(current);
current=current->left;
}
current = myStack.top();
myStack.pop();
result.push_back(current->val);
if(current->right!=NULL)
current = current->right;
else
current = NULL;
}
return result;
}

后续遍历：

/*
后续遍历需要用一个lastVisited节点来记录刚刚处理过的节点。按照后续遍历的顺序来遍历就可以了。
因为当右孩子为空时他就是当前那个小二叉树的左叶子节点，当他的右孩子为lastVisited节点之后他
就是被处理完的子树的父节点，也要进行处理。所以可以根据这个写出如下代码。
*/
vector<int> postorderTraversal(TreeNode *root) {
vector<int> result;
stack<TreeNode *> myStack;

TreeNode *current = root, *lastVisited = NULL;
while (current != NULL || !myStack.empty()) {
while (current != NULL) {
myStack.push(current);
current = current->left;
}
current = myStack.top();
if (current->right == NULL || current->right == lastVisited) {
myStack.pop();
result.push_back(current->val);
lastVisited = current;
current = NULL;
} else {
current = current->right;
}
}
return result;

总结下一些写代码的思路：

1.想清楚处理这问题的思路。开始状态–处理过程–结束状态。

2.在处理数据之前要先确保数据是可以被处理的，比如在处理数组的时候要先检查是否越界。

3.处理一次之后如何进行下一次处理一定一定要搞清楚。就像上述的迭代遍历二叉树的方法，如果最后不将current设置为NULL那就肯定会出问题。他会无限重复的进行循环。