C++_学习随笔_重构二叉树和判断是否是对称二叉树

通过输入二叉树的前序遍历和中序遍历重构二叉树，

然后对二叉树进行是否是对称二叉树的判断。

#include<iostream>
#include<vector>
using namespace std;

struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x):val(x),left(NULL),right(NULL){}
};

//根据前序遍历和中序遍历重构二叉树
TreeNode* reConstructBinaryTree(vector<int>pre, vector<int>vin) {
    if (pre.empty() || vin.empty())
        return nullptr;
    //根节点
    TreeNode *root = new TreeNode(pre[0]);
    int root_index = 0;
    //找到根节点的索引值
    for (int i = 0; i < pre.size(); i++) {
        if (vin[i] == pre[0]) {
            root_index = i;
            break;
        }
    }
    vector<int> pre_left, pre_right, vin_left, vin_right;
    for (int i = 0; i < root_index; i++) {
        pre_left.push_back(pre[i + 1]);
        vin_left.push_back(vin[i]);
    }

    for (int i = root_index + 1; i < pre.size(); i++) {
        pre_right.push_back(pre[i]);
        vin_right.push_back(vin[i]);
    }

    root->left = reConstructBinaryTree(pre_left, vin_left);
    root->right = reConstructBinaryTree(pre_right, vin_right);

    return root;
}


//判断是否是对称的
bool isSymmetrical(TreeNode* p1, TreeNode* p2) {
    if (p1 == NULL && p2 == NULL)
        return true;
    if (p1 == NULL || p2 == NULL)
        return false;
    if (p1->val != p2->val)
        return false;
    return isSymmetrical(p1->left, p2->right) && isSymmetrical(p1->right, p2->left);
}


bool isSymmetrical(TreeNode* pRoot) {
    if (pRoot == NULL) {
        return true;
    }
    return isSymmetrical(pRoot, pRoot);
}


int main() {
    vector<int> pre, vin;
    cout << "请输入节点数："<<endl;
    int n = 0,temp;
    cin >> n;
    cout << "请输入前序遍历：" << endl;
    for (int i = 0; i < n; i++) {
        cin >> temp;
        pre.push_back(temp);
    }
    cout << "请输入中序遍历：" << endl;
    for (int i = 0; i < n; i++) {
        cin >> temp;
        vin.push_back(temp);
    }
    bool is;
    TreeNode* root_node = reConstructBinaryTree(pre, vin);
    is = isSymmetrical(root_node);
    cout << is;
    
}

　　　　　　　　8

　　　　6　　　　　　　　6

5　　　　　　7　　7　　　　　　　　5

 

　　　　　　　　　　8

　　　　　　6　　　　　　　　9

　　5　　　　　　7　　7　　　　　　5

 

参考链接：

https://www.cnblogs.com/wanglei5205/p/8503036.html

https://cuijiahua.com/blog/2018/01/basis_58.html