我在代码随想录|写代码Day15之二叉树遍历

思路

这次我们要好好谈一谈递归，为什么很多同学看递归算法都是“一看就会，一写就废”。

主要是对递归不成体系，没有方法论，每次写递归算法 ，都是靠玄学来写代码，代码能不能编过都靠运气。

本篇将介绍前后中序的递归写法，一些同学可能会感觉很简单，其实不然，我们要通过简单题目把方法论确定下来，有了方法论，后面才能应付复杂的递归。

这里帮助大家确定下来递归算法的三个要素。每次写递归，都按照这三要素来写，可以保证大家写出正确的递归算法！

1. 确定递归函数的参数和返回值： 确定哪些参数是递归的过程中需要处理的，那么就在递归函数里加上这个参数， 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。

2. 确定终止条件： 写完了递归算法, 运行的时候，经常会遇到栈溢出的错误，就是没写终止条件或者终止条件写的不对，操作系统也是用一个栈的结构来保存每一层递归的信息，如果递归没有终止，操作系统的内存栈必然就会溢出。

3. 确定单层递归的逻辑： 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。

好了，我们确认了递归的三要素，接下来就来练练手：

以下以前序遍历为例：

1. 确定递归函数的参数和返回值：因为要打印出前序遍历节点的数值，所以参数里需要传入vector来放节点的数值，除了这一点就不需要再处理什么数据了也不需要有返回值，所以递归函数返回类型就是void，代码如下：

void traversal(TreeNode* cur, vector<int>& vec)

1. 确定终止条件：在递归的过程中，如何算是递归结束了呢，当然是当前遍历的节点是空了，那么本层递归就要结束了，所以如果当前遍历的这个节点是空，就直接return，代码如下：

if (cur == NULL) return;

1. 确定单层递归的逻辑：前序遍历是中左右的循序，所以在单层递归的逻辑，是要先取中节点的数值，代码如下：

vec.push_back(cur->val);    // 中
traversal(cur->left, vec);  // 左
traversal(cur->right, vec); // 右

单层递归的逻辑就是按照中左右的顺序来处理的，这样二叉树的前序遍历，基本就写完了，再看一下完整代码：

1.前序遍历 144. 二叉树的前序遍历

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
//前序遍历
  vector<int> t(TreeNode* cur) {  
    vector<int> ans;  
    if (cur == nullptr) return ans;  
    ans.push_back(cur->val);  
    // 合并左子树的遍历结果  
    vector<int> left = t(cur->left);  
    ans.insert(ans.end(), left.begin(), left.end());  
    // 合并右子树的遍历结果  
    vector<int> right = t(cur->right);  
    ans.insert(ans.end(), right.begin(), right.end());  
    return ans; // 返回合并后的结果  
}
    vector<int> preorderTraversal(TreeNode* root) {//一个根结点
        TreeNode*cur = root;
        return t(cur);
    }
}

2.中序遍历 :

代码1

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
//前序遍历
  vector<int> t(TreeNode* cur) {  
    vector<int> ans;  
    if (cur == nullptr) return ans;  
    // 合并左子树的遍历结果  
    vector<int> left = t(cur->left);  
    ans.insert(ans.end(), left.begin(), left.end());  
    ans.push_back(cur->val);  
    // 合并右子树的遍历结果  
    vector<int> right = t(cur->right);  
    ans.insert(ans.end(), right.begin(), right.end());  
    return ans; // 返回合并后的结果  
}
    vector<int> inorderTraversal(TreeNode* root) {//一个根结点
        TreeNode*cur = root;
        return t(cur);
    }
};

 代码2

class Solution {
private: 
    void dfs(TreeNode* node, vector<int>& res){
        if(!node)return;
        dfs(node->left, res);     // 先处理左子树
        res.emplace_back(node->val);  // 再处理当前节点
        dfs(node->right, res);    // 最后处理右子树
    }

public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        dfs(root, res);
        return res;
    }
};

后序遍历  145. 二叉树的后序遍历

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
//后序遍历
  vector<int> t(TreeNode* cur) {  
    vector<int> ans;  
    if (cur == nullptr) return ans;  
    // 合并左子树的遍历结果  
    vector<int> left = t(cur->left);  
    ans.insert(ans.end(), left.begin(), left.end());   
    // 合并右子树的遍历结果  
    vector<int> right = t(cur->right);  
    ans.insert(ans.end(), right.begin(), right.end());  
    ans.push_back(cur->val); 
    return ans; // 返回合并后的结果  
}
    vector<int> postorderTraversal(TreeNode* root) {
        TreeNode*cur = root;
        return t(cur);
    }
};

三种遍历的非递归写法

// 不使用递归
 typedef struct TreeNode TreeNode;
 // 获取节点数
int treeGetNumOfNodes(TreeNode* root)
{
    if(root == NULL)
    {
        return 0;
    }
    else 
    {
        return 1+treeGetNumOfNodes(root->left)+treeGetNumOfNodes(root->right);
    }
}

// 定义栈
typedef struct{
    int size;          // 栈数组的长度
    TreeNode** array;  // 栈的元素是树节点指针
    int top;
}mystack;

mystack* stackCreat(int maxsize)
{
    mystack* newstack = (mystack*)malloc(sizeof(mystack));
    newstack->size = maxsize;
    newstack->top = 0;
    newstack->array = (TreeNode**)malloc(sizeof(TreeNode*)*maxsize);
    return newstack;
}

void stackPop(mystack*stk)
{
    if(stk->top>0)
    {
        --stk->top;
    }
}

void stackPush(mystack*stk,TreeNode* val)
{
    if(stk->top<stk->size)
    {
        stk->array[stk->top++] = val;
    }
}

int stackIsEmpty(mystack*stk)
{
    return stk->top == 0;
}

TreeNode* stackGetTopVal(mystack*stk)
{
    if(stackIsEmpty(stk))
    {
        return NULL;
    }
    return stk->array[stk->top-1];
}

void stackFree(mystack*stk)
{
    if(stk->size>0)
    {
        free(stk->array);
    }
    free(stk);
    stk = NULL;
}

#define MAX_STACK_SIZE       100
#define PREORDER        1
#define INORDER         2
#define POSTORDER       3
void traversal(TreeNode*root,int*ret,int*index,int traversalType)
{
    if(root==NULL)
    {
        return;
    }
    mystack*stk = stackCreat(MAX_STACK_SIZE);
    stackPush(stk,root);
    while(!stackIsEmpty(stk))
    {
        TreeNode*p = stackGetTopVal(stk);
        stackPop(stk);    // 出栈
        if(p!=NULL)
        {
            switch(traversalType){
                case INORDER:
                {
                    // 中序遍历
                    if(p->right) stackPush(stk, p->right);
                    stackPush(stk, p);
                    stackPush(stk, NULL);
                    if(p->left)  stackPush(stk, p->left);
                    break;
                }
                case POSTORDER:
                {
                    // 后序遍历
                    stackPush(stk, p);
                    stackPush(stk, NULL);
                    if(p->right) stackPush(stk, p->right);
                    if(p->left)  stackPush(stk, p->left);
                    break;
                }
                default:
                {
                    // 先序遍历，输入错误时默认选择
                    if(p->right) stackPush(stk, p->right);
                    if(p->left)  stackPush(stk, p->left);
                    stackPush(stk, p);
                    stackPush(stk, NULL);
                }
            }
        }
        else
        {
            // 节点已经访问过，读取其数据
            p = stackGetTopVal(stk);
            stackPop(stk);
            ret[*index] = p->val;
            (*index)++;
        }    
    }
    stackFree(stk);
}


int* postorderTraversal(struct TreeNode* root, int* returnSize){
    int len = treeGetNumOfNodes(root);
    *returnSize = len;
    int *ret = (int*)malloc(sizeof(int)*len);
    int index = 0;
    // 后序遍历
    traversal(root,ret,&index,POSTORDER);
    return ret;
}