当前位置：首页 > article >正文

2、设计在链式存储结构上交换二叉树中所有结点左右子树的算法。

article 2024/9/23 17:14:49

要在链式存储结构上交换二叉树中所有节点的左右子树，你可以采用递归的方式。对于每个节点，交换其左右子树，并递归地对左子树和右子树执行相同的操作。

#include <iostream>

struct TreeNode {
    int data;
    TreeNode* left;
    TreeNode* right;
    //构造，可根据需要删除
    TreeNode(int val) : data(val), left(nullptr), right(nullptr) {}
};

//核心交换代码
void swapLeftAndRight(TreeNode* root) {
    if (root == nullptr) {
        return;
    }

    // 递归交换左右子树
    swapLeftAndRight(root->left);
    swapLeftAndRight(root->right);

    // 交换左右子树的指针
    TreeNode* temp = root->left;
    root->left = root->right;
    root->right = temp;
}

// 验证交换结果：中序遍历
void inorderTraversal(TreeNode* root) {
    if (root != nullptr) {
        inorderTraversal(root->left);
        std::cout << root->data << " ";
        inorderTraversal(root->right);
    }
}

int main() {
    // 构建一个简单的二叉树作为例子
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5); 
    inorderTraversal(root); 
    printf("\n");
    // 交换左右子树
    swapLeftAndRight(root);
    inorderTraversal(root); 
    return 0;
}