算法训练day20Leetcode654最大二叉树617合并二叉树700二叉树中的1搜索98验证二叉搜索树

dc爱傲雪和技术2024-01-21 23:56

654 最大二叉树

题目

https://leetcode.cn/problems/maximum-binary-tree/description/

我的想法

中序遍历递归，找到最大值然后作为根节点

题目分析

凡是构造二叉树的题目都用前序遍历 （中左右）

为先构造中间节点，然后递归构造左子树和右子树。

确定递归函数的参数和返回值

参数传入的是存放元素的数组，返回该数组构造的二叉树的头结点，返回类型是指向节点的指针。

1.确定终止条件

题目中说了输入的数组大小一定是大于等于1的，所以我们不用考虑小于1的情况，那么当递归遍历的时候，如果传入的数组大小为1，说明遍历到了叶子节点了。

那么应该定义一个新的节点，并把这个数组的数值赋给新的节点，然后返回这个节点。 这表示一个数组大小是1的时候，构造了一个新的节点，并返回。

TreeNode* node = new TreeNode(0);
if (nums.size() == 1) {
    node->val = nums[0];
    return node;
}

确定单层递归逻辑

这里有三步工作

先要找到数组中最大的值和对应的下标， 最大的值构造根节点，下标用来下一步分割数组。

最大值所在的下标左区间 构造左子树

这里要判断maxValueIndex > 0，因为要保证左区间至少有一个数值。

最大值所在的下标右区间 构造右子树

判断maxValueIndex < (nums.size() - 1)，确保右区间至少有一个数值。

acm模式代码

cpp 复制代码 
#include <iostream>
#include <vector>
struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x):val(x), left(nullptr), right(nullptr) {}
    TreeNode(int x, TreeNode* left, TreeNode* right):val(x), left(left), right(right) {}

};

class Solution {
public:
    TreeNode* constrcuctMaxinumBinaryTree(std::vector<int>& num) {
        TreeNode* node = new TreeNode(0);
        if (num.size() == 1) {
            node->val = num[0];
            return node;
        }
        int maxvalue = 0;
        int maxvalueIndex = 0;
        for (int i = 0; i < num.size(); i++) {
            if (num[i] > maxvalue) {
                maxvalue = num[i];
                maxvalueIndex = i;
            }
        }
        node->val = maxvalue;
        if(maxvalueIndex > 0) {
            std::vector<int> newVec(num.begin(), num.begin() + maxvalueIndex);
            node->left = constrcuctMaxinumBinaryTree(newVec);
        }
        if (maxvalueIndex < num.size()-1) {
            std::vector<int> newVec(num.begin() + maxvalueIndex + 1, num.end());
            node->right = constrcuctMaxinumBinaryTree(newVec);
        }
        return node;
    }
};

void printTree(TreeNode* node) {
    if (node != nullptr) {
        printTree(node->left);
        std::cout << node->val << " ";
        printTree(node->right);
    }
}

int main() {
    // 创建一个整数向量
    std::vector<int> nums = {3, 2, 1, 6, 0, 5};

    // 创建Solution实例并构建最大二叉树
    Solution solution;
    TreeNode* root = solution.constrcuctMaxinumBinaryTree(nums);

    // 打印构建的二叉树
    std::cout << "The constructed maximum binary tree (inorder): ";
    printTree(root);
    std::cout << std::endl;

    // 注意：这里没有删除TreeNode的实例，实际应用中应考虑内存管理

    return 0;
}

617 合并二叉树

题目描述

https://leetcode.cn/problems/merge-two-binary-trees/description/

我的想法

还是构造新的二叉树，选择前序遍历

题目分析

给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点。

我们来按照递归三部曲来解决：

  1. 确定递归函数的参数和返回值：
    首先要合入两个二叉树，那么参数至少是要传入两个二叉树的根节点，返回值就是合并之后二叉树的根节点。

代码如下：

TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
  1. 确定终止条件：
    因为是传入了两个树，那么就有两个树遍历的节点t1 和 t2，如果t1 == NULL 了，两个树合并就应该是 t2 了（如果t2也为NULL也无所谓，合并之后就是NULL）。

反过来如果t2 == NULL，那么两个数合并就是t1（如果t1也为NULL也无所谓，合并之后就是NULL）。

代码如下：

if (t1 == NULL) return t2; // 如果t1为空，合并之后就应该是t2
if (t2 == NULL) return t1; // 如果t2为空，合并之后就应该是t1
  1. 确定单层递归的逻辑
    单层递归的逻辑就比较好写了，这里我们重复利用一下t1这个树，t1就是合并之后树的根节点（就是修改了原来树的结构）。

那么单层递归中，就要把两棵树的元素加到一起。

t1->val += t2->val;

接下来t1 的左子树是：合并 t1左子树 t2左子树之后的左子树。

t1 的右子树：是 合并 t1右子树 t2右子树之后的右子树。

最终t1就是合并之后的根节点。

代码如下：

t1->left = mergeTrees(t1->left, t2->left);
t1->right = mergeTrees(t1->right, t2->right);
return t1;

acm完整代码

cpp 复制代码 
#include <iostream>

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

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if (root1 == nullptr) return root2;
        if (root2 == nullptr) return root1;
        TreeNode* root = new TreeNode(0);

        root->val = root1->val + root2->val;
        root->left = mergeTrees(root1->left, root2->left);
        root->right = mergeTrees(root1->right, root2->right);
        return root;
    }
};

void printTree(TreeNode* node) {
    if (node != nullptr) {
        std::cout << node->val << " ";

        printTree(node->left);
        printTree(node->right);
    }
}

int main() {
    // 创建两个树的根节点
    TreeNode* root1 = new TreeNode(1);
    root1->left = new TreeNode(3);
    root1->right = new TreeNode(2);
    root1->left->left = new TreeNode(5);

    TreeNode* root2 = new TreeNode(2);
    root2->left = new TreeNode(1);
    root2->right = new TreeNode(3);
    root2->left->right = new TreeNode(4);
    root2->right->right = new TreeNode(7);

    // 合并两棵树
    Solution solution;
    TreeNode* mergedTree = solution.mergeTrees(root1, root2);

    // 打印合并后的树
    std::cout << "The merged tree (inorder): ";
    printTree(mergedTree);
    std::cout << std::endl;

    // 注意：这里没有删除TreeNode的实例，实际应用中应考虑内存管理
    delete root1;
    delete root2;
    delete mergedTree;
    return 0;
}

700二叉搜索树中的搜索

题目描述

https://leetcode.cn/problems/search-in-a-binary-search-tree/description/

我的想法

直接用前序遍历，遍历到值相等的节点就作为根节点返回

题目分析

二叉搜索树是一个有序树：

若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值；

若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；

它的左、右子树也分别为二叉搜索树

这就决定了，二叉搜索树，递归遍历和迭代遍历和普通二叉树都不一样。

本题，其实就是在二叉搜索树中搜索一个节点。那么我们来看看应该如何遍历。

递归法

  1. 确定递归函数的参数和返回值
    递归函数的参数传入的就是根节点和要搜索的数值，返回的就是以这个搜索数值所在的节点。

代码如下：

TreeNode* searchBST(TreeNode* root, int val)

  1. 确定终止条件
    如果root为空，或者找到这个数值了，就返回root节点。

    if (root == NULL || root->val == val) return root;

  2. 确定单层递归的逻辑
    看看二叉搜索树的单层递归逻辑有何不同。

因为二叉搜索树的节点是有序的，所以可以有方向的去搜索。

如果root->val > val，搜索左子树，如果root->val < val，就搜索右子树，最后如果都没有搜索到，就返回NULL。

代码如下：

TreeNode* result = NULL;
if (root->val > val) result = searchBST(root->left, val);
if (root->val < val) result = searchBST(root->right, val);
return result;

acm模式代码

cpp 复制代码 
class Solution{
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if (root == nullptr) return nullptr;
        if (root->val == val) return root;

        TreeNode* leftSearch = searchBST(root->left, val);
        if (leftSearch != nullptr) return leftSearch;

        return searchBST(root->right, val);
    }

    void printTree(TreeNode* root) {
        if (root == nullptr) return;
        std::cout << root->val << " ";
        printTree(root->left);
        printTree(root->right);
    }
};

int main() {
    TreeNode* node = new TreeNode(3);
    node->left = new TreeNode(2);
    node->right = new TreeNode(3);
    node->left->left = new TreeNode(4);
    node->left->right = new TreeNode(5);

    Solution* sol = new Solution();
    TreeNode* newnode = sol->searchBST(node, 2);
    sol->printTree(newnode);

    // 注意：这里没有删除TreeNode的实例和Solution的实例，实际应用中应考虑内存管理

    return 0;
}

98验证二叉搜索树

题目描述

https://leetcode.cn/problems/validate-binary-search-tree/description/

我的想法

采用后序遍历，比较两个儿子是否满足条件。

题目分析

二叉搜索类题目中序遍历是有序的

有了这个特性，验证二叉搜索树，就相当于变成了判断一个序列是不是递增的了。

可以递归中序遍历将二叉搜索树转变成一个数组，代码如下：

vector<int> vec;
void traversal(TreeNode* root) {
    if (root == NULL) return;
    traversal(root->left);
    vec.push_back(root->val); // 将二叉搜索树转换为有序数组
    traversal(root->right);
}
然后只要比较一下，这个数组是否是有序的，注意二叉搜索树中不能有重复元素。

traversal(root);
for (int i = 1; i < vec.size(); i++) {
    // 注意要小于等于，搜索树里不能有相同元素
    if (vec[i] <= vec[i - 1]) return false;
}
return true;

完整acm模式代码

cpp 复制代码 
#include <iostream>
#include <vector>

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

class Solution{
public:
    std::vector<int> vec;

    void traversal(TreeNode* root) {
        if (root == nullptr) return;
        traversal(root->left);
        vec.push_back(root->val);
        traversal(root->right);
        
    }

    bool isValidBST(TreeNode* root) {
        traversal(root);
        for (int i = 1; i < vec.size(); i++) {
            if (vec[i] <= vec[i - 1]) return false;
        }
        return true;
    }
};

int main() {
    // Create nodes
    TreeNode* n1 = new TreeNode(2);
    TreeNode* n2 = new TreeNode(1);
    TreeNode* n3 = new TreeNode(3);

    // Construct the tree
    n1->left = n2;
    n1->right = n3;

    // Create a solution instance
    Solution solution;

    // Check if the tree is a valid BST
    bool result = solution.isValidBST(n1);
    if (result) {
        std::cout << "The tree is a valid BST." << std::endl;
    } else {
        std::cout << "The tree is not a valid BST." << std::endl;
    }

    // Clean up
    delete n1;
    delete n2;
    delete n3;

    return 0;
}

今日学习链接

https://www.bilibili.com/video/BV1wG411g7sF

https://programmercarl.com/0098.验证二叉搜索树.html

相关推荐
烦躁的大鼻嘎5 分钟前
模拟算法实例讲解：从理论到实践的编程之旅
数据结构·c++·算法·leetcode
C++忠实粉丝22 分钟前
计算机网络socket编程(4)_TCP socket API 详解
网络·数据结构·c++·网络协议·tcp/ip·计算机网络·算法
用户37791362947551 小时前
【循环神经网络】只会Python，也能让AI写出周杰伦风格的歌词
人工智能·算法
福大大架构师每日一题1 小时前
文心一言 VS 讯飞星火 VS chatgpt （396）-- 算法导论25.2 1题
算法·文心一言
EterNity_TiMe_1 小时前
【论文复现】（CLIP）文本也能和图像配对
python·学习·算法·性能优化·数据分析·clip
机器学习之心1 小时前
一区北方苍鹰算法优化+创新改进Transformer！NGO-Transformer-LSTM多变量回归预测
算法·lstm·transformer·北方苍鹰算法优化·多变量回归预测·ngo-transformer
yyt_cdeyyds1 小时前
FIFO和LRU算法实现操作系统中主存管理
算法
alphaTao2 小时前
LeetCode 每日一题 2024/11/18-2024/11/24
算法·leetcode
kitesxian2 小时前
Leetcode448. 找到所有数组中消失的数字(HOT100)+Leetcode139. 单词拆分(HOT100)
数据结构·算法·leetcode
VertexGeek3 小时前
Rust学习（八）：异常处理和宏编程：
学习·算法·rust
热门推荐
01(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明02PyTorch机器学习实现液态神经网络03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04玄机平台应急响应—webshell查杀05Coze扣子平台完整体验和实践（附国内和国际版对比）06Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO07【目标跟踪】相机运动补偿08Unity中PICO实现 隔空取物 和 接触抓取物体09RAG 实践- Ollama+RagFlow 部署本地知识库10量化方法怎么选？如何评估量化后的大模型LLM？