迭代遍历二叉树

今天想着学动态规划，不过发现貌似有点难度，功力暂时还不够。然后看到了一个二叉树迭代遍历方法的题目，感觉挺有意思的，就写了个。
通过这个可以总结的写代码的经验。先上代码吧。
前序：

/*
    前序遍历其实是比较简单的，因为递归的实现本来就是用的栈的原理，所以直接用栈来写二叉树的遍历就好。
    前序遍历就是第一次遇到就把遇到的数据处理。所以就遇到就输出，一直到最左叶子节点。
    然后开始处理右边的节点，右子树为空继续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那就肯定会出问题。他会无限重复的进行循环。