二叉树结构及其遍历

力扣上对二叉树的链式代码实现：

1
2
3
4
5
6
7
8
9
//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) {}
  };

递归遍历

基本模板:

1
2
3
4
5
6
7
8
void RecursionTraverse(TreeNode *root ){
    if(!root) return;
    //在此插入前序遍历位置的代码
    RecursionTraverse(root->left);
    //中序遍历位置
    RecursionTraverse(root->right); 、
    //后序遍历位置
}

效果：进入根节点后一直往左节点走，直到遇到空指针后，返回上一层的父节点并执行父节点的RecursionTraverse(root->right)并以此往复。

递归遍历示意图

Note：递归遍历节点的顺序仅取决于左右子节点的递归调用顺序，与其他代码无关。

与所谓的前、中、后序遍历的区别是，前中后序遍历的结果不同，原因是因为把代码写在了不同位置，所以产生了不同的效果。而这与递归函数（上述模板）本身如何实现没有关系。

总结：在两行RecursionTraverse递归调用的相对顺序不变的情况下,函数本身访问节点的顺序不变，而在前、中、后序插入代码位置改变的是对应代码的执行顺序。

By the way,个人认为理清这个递归遍历的过程能够很好地帮助理解递归算法，

前、中、后顺序对应的也就是

•代码在刚进入节点（还未进行进一步递归）

•节点的递归完成一半(左子树已遍历完成)

•节点的递归完成（左右字树都已完成，准备弹出当前节点的值）

这三个时机

层序遍历

对于初学者来说,层序遍历的顺序比较直观，即在一层从左到右遍历，直到遍历完后再进入下一层，以此往复。

基本模板:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
void LevelOrderTraverse(TreeNode *root){
    if(!root) return;
    queue<TreeNode*> q;
    q.push(root);
    int depth = 1;
    
    while(!q.empty()){
        int sz=q.size();//用于记录每层的节点数量
        //每次pop完当前层的节点后退出内层while循环，回到int sz=q.size()更新为下一层的节点数量
        while(sz-- > 0){
            TreeNode* cur =q.front();
            q.pop(); 
            //cout << "depth = " << depth << ", val = " << cur->val << endl;
            if(cur->left != nullptr) q.push(cur->left);
            if(cur->right != nullptr) q.push(cur->right);
        }
        depth++;
    }
}

如果涉及到每条树枝的权重不同，要求遍历后输出每个节点对应的路径权重和，则不能仅设置一个“全局”的depth变量了。

此时可以创建一个新的类State：

1
2
3
4
5
6
7
class State{
public:
    TreeNode* node;
    int depth;
    
    State(TreeNode* node,int depth): node(node),depth(depth){}
};

然后队列从存储TreeNode指针改为存储State，以便每个节点单独维护自己的路径权重和即可，其余代码逻辑基本不变。