好的,这是二叉树中序线索化的代码:
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
// 新增的线索化属性
bool is_thread; // 是否为线索
TreeNode *prev; // 前驱结点
TreeNode *next; // 后继结点
TreeNode(int x) : val(x), left(nullptr), right(nullptr), is_thread(false), prev(nullptr), next(nullptr) {}
};
void inorder_thread(TreeNode *root) {
if (!root) return;
static TreeNode *prev = nullptr; // 前驱结点
inorder_thread(root->left);
if (!root->left) { // 如果没有左孩子,则指向前驱结点
root->is_thread = true;
root->prev = prev;
}
if (prev && !prev->right) { // 如果前驱结点没有右孩子,则指向当前结点
prev->is_thread = true;
prev->next = root;
}
prev = root;
inorder_thread(root->right);
}
这段代码实现了二叉树中序线索化的过程。具体来说,首先递归遍历左子树,然后对当前结点进行线索化。如果当前结点没有左孩子,则指向前驱结点。如果前驱结点没有右孩子,则指向当前结点。最后递归遍历右子树。
线索化后,二叉树就变成了一个双向链表,可以方便地遍历整棵树。