洛谷P3365 改造二叉树：从问题分析到代码实现

3天前洛谷题解66

一、问题分析

题目要求我们计算将二叉树修改为二叉搜索树(BST)所需的最少修改次数。二叉搜索树的性质是：对于任意节点，其左子树所有节点的值都小于该节点的值，右子树所有节点的值都大于该节点的值。

二、解题思路

‌中序遍历序列‌：BST的中序遍历结果是一个严格递增序列
‌问题转化‌：将原二叉树的中序遍历序列转换为严格递增序列所需的最少修改次数
‌最长递增子序列(LIS)‌：最少修改次数 = 序列长度 - 最长递增子序列长度

三、C++代码实现

#include <iostream>
#include <vector>
#include <algorithm>
using namespACe std;

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

// 构建二叉树
TreeNode* buildTree(int n, const vector<int>& vals, const vector<pair<int, int>>& edges) {
    vector<TreeNode*> nodes(n + 1);
    for (int i = 1; i <= n; ++i) {
        nodes[i] = new TreeNode(vals[i - 1]);
    }
    
    for (int i = 2; i <= n; ++i) {
        int fa = edges[i - 2].first;
        int ch = edges[i - 2].second;
        if (ch == 0) {
            nodes[fa]->left = nodes[i];
        } else {
            nodes[fa]->right = nodes[i];
        }
    }
    return nodes[1];
}

// 中序遍历收集节点值
void inorder(TreeNode* root, vector<int>& seq) {
    if (!root) return;
    inorder(root->left, seq);
    seq.push_back(root->val);
    inorder(root->right, seq);
}

// 计算最长递增子序列长度
int lengthOfLIS(vector<int>& nums) {
    vector<int> dp;
    for (int num : nums) {
        auto it = lower_bound(dp.begin(), dp.end(), num);
        if (it == dp.end()) {
            dp.push_back(num);
        } else {
            *it = num;
        }
    }
    return dp.size();
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int n;
    cin >> n;
    
    vector<int> vals(n);
    for (int i = 0; i < n; ++i) {
        cin >> vals[i];
    }
    
    vector<pair<int, int>> edges(n - 1);
    for (int i = 0; i < n - 1; ++i) {
        cin >> edges[i].first >> edges[i].second;
    }
    
    TreeNode* root = buildTree(n, vals, edges);
    vector<int> seq;
    inorder(root, seq);
    
    int lis_len = lengthOfLIS(seq);
    cout << n - lis_len << endl;
    
    return 0;
}

四、代码详解

1 数据结构定义

我们首先定义树节点的结构：

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

2 构建二叉树

根据输入构建二叉树：

TreeNode* buildTree(int n, const vector<int>& vals, const vector<pair<int, int>>& edges) {    vector<TreeNode*> nodes(n + 1);    for (int i = 1; i <= n; ++i) {
        nodes[i] = new TreeNode(vals[i - 1]);
    }    
    for (int i = 2; i <= n; ++i) {        int fa = edges[i - 2].first;        int ch = edges[i - 2].second;        if (ch == 0) {
            nodes[fa]->left = nodes[i];
        } else {
            nodes[fa]->right = nodes[i];
        }
    }    return nodes[1];
}

3 中序遍历

收集中序遍历序列：

void inorder(TreeNode* root, vector<int>& seq) {    if (!root) return;    inorder(root->left, seq);
    seq.push_back(root->val);    inorder(root->right, seq);
}

4 计算最长递增子序列

使用贪心算法计算LIS长度：

int lengthOfLIS(vector<int>& nums) {
    vector<int> dp;    for (int num : nums) {        auto it = lower_bound(dp.begin(), dp.end(), num);        if (it == dp.end()) {
            dp.push_back(num);
        } else {
            *it = num;
        }
    }    return dp.size();
}