leetcode 99 回复二叉搜索树 C语言实现

测评

题目

leetcode 99
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下，恢复这棵树。

示例 1:
输入: [1,3,null,null,2]

1
/
3

2

输出: [3,1,null,null,2]
3
/
1

2

示例 2:
输入: [3,1,4,null,null,2]

3
/
1 4
/
2

输出: [2,1,4,null,null,3]
2
/
1 4
/
3

代码

C语言实现，找到两个错误节点，再把其值交换

void visit(struct TreeNode *root, struct TreeNode **pre, struct TreeNode **p, struct TreeNode **q, int *count) {
    if (root == NULL) return ;
    visit(root->left, pre, p, q, count);
    if (*count == 0 && *pre != NULL && (*pre)->val > root->val) {
        (*count) = 1;
        *p = *pre;
    }
    if (*count == 1 && *pre != NULL && (*pre)->val > root->val) {
        *q = root;
    }
    *pre = root;
    visit(root->right, pre, p, q, count);
}
void recoverTree(struct TreeNode* root){
    struct TreeNode *p = NULL, *q = NULL, *pre = NULL;
    int count = 0;
    visit(root, &pre, &p, &q, &count);
    int t = p->val;
    p->val = q->val;
    q->val = t;
    return ;
}