保研机试之【三道二叉树习题，思路为主】

建议先过一遍：今天是二叉树~-CSDN博客，very重要！

第一题：226. 翻转二叉树 - 力扣（LeetCode）

回溯写法：（前中后位置交换都可以）

/**
 * 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:
    void traverse(TreeNode* root){
        if(root==nullptr){return;}
        //前
        TreeNode* temp=root->left;
        root->left=root->right;
        root->right=temp;
        traverse(root->left);
        //中
        traverse(root->right);
        //后
    }
    TreeNode* invertTree(TreeNode* root) {
        traverse(root);
        return root;
    }
};

动规写法：（后序位置闷声干大事--交换）

/**
 * 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:
    TreeNode* invertTree(TreeNode* root) {
        if(root==nullptr){return nullptr;}
        //前
        TreeNode* left=invertTree(root->left);
        //中
        TreeNode* right=invertTree(root->right);
        //后
        root->left=right;
        root->right=left;
        return root;
    }
};

第二题：116. 填充每个节点的下一个右侧节点指针 - 力扣（LeetCode）

第一眼是层次遍历，但是不让用队列，，

回溯写法：(感觉很难想，将相邻节点之间的空隙看做一个整体）

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node* _left, Node* _right, Node* _next)
        : val(_val), left(_left), right(_right), next(_next) {}
};
*/

class Solution {
public:
    void traverse(Node* left,Node* right){
        if(left==nullptr||right==nullptr){return;}
        //前
        left->next=right;
        traverse(left->left,left->right);
        traverse(right->left,right->right);
        traverse(left->right,right->left);
    }
    Node* connect(Node* root) {
        if(root==nullptr){return nullptr;}
        traverse(root->left,root->right);
        return root;
    }
};

再来一道有一丢丢类似的：101. 对称二叉树 - 力扣（LeetCode） 

/**
 * 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:
    bool res=true;
    void traverse(TreeNode* left,TreeNode* right){
        if((left==nullptr && right!=nullptr) || (left!=nullptr &&right==nullptr)){res=false;return;}
        if(left==nullptr && right==nullptr){return;}
        if(left->val!=right->val){res=false;return;}
        traverse(left->left,right->right);
        traverse(left->right,right->left);
    }
    bool isSymmetric(TreeNode* root) {
        if(root==nullptr){return true;}
        traverse(root->left,root->right);
        if(res==true){return true;}
        else{return false;}
    }
};

第三题：114. 二叉树展开为链表 - 力扣（LeetCode）

耶耶耶，自己做出来的！

采用的是动规写法，返回的左子树是已经排好的，返回的右子树也是已经排好的，只要把左子树最右下角的右指针指向右子树即可。

/**
 * 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:
    TreeNode* fla(TreeNode* root){
        if(root==nullptr){
            return nullptr;
        }
        //前
        TreeNode* left=fla(root->left);
        //中
        TreeNode* right=fla(root->right);
        //后
        TreeNode* temp=right;
        root->right=left;
        root->left=nullptr;
        TreeNode* last=root;
        while(last->right){
            last=last->right;
        }
        last->right=temp;
        return root;
    }
    void flatten(TreeNode* root) {
        fla(root);
    }
};

参考：东哥带你刷二叉树（思路篇） | labuladong 的算法笔记