算法训练营打卡Day18

idris的算法日记

于 2024-09-29 23:28:53 发布

阅读量581

点赞数 11

文章标签：算法数据结构 c++ python 深度优先广度优先

本文链接：https://blog.csdn.net/Zoe_igxgx/article/details/142644950

版权

题目1、二叉搜索树的最小绝对差

力扣题目链接(opens new window)

给你一棵所有节点为非负值的二叉搜索树，请你计算树中任意两节点的差的绝对值的最小值。

示例：

530二叉搜索树的最小绝对差

思路

遇到在二叉搜索树上求什么最值或者差值之类的问题，我们可以尝试把它想成在一个有序数组上求解。本题使用可以递归的方法，把二叉搜索树转换成有序数组，然后遍历一遍数组，从而求解最小绝对值差。

代码实现

python

class Solution:
    def __init__(self):
        self.vec = []

    def traversal(self, root):
        if root is None:
            return
        self.traversal(root.left)  #左
        self.vec.append(root.val)  #中  将二叉搜索树转换为有序数组
        self.traversal(root.right) #右

    def getMinimumDifference(self, root):
        self.vec = []              #清空数组
        self.traversal(root)
        if len(self.vec) < 2:      #遇到节点数少于2的二叉树
            return 0
        result = float('inf')
        for i in range(1, len(self.vec)):
            # 统计有序数组的最小差值
            result = min(result, self.vec[i] - self.vec[i - 1])
        return result

C++

class Solution {
private:
vector<int> vec;
void traversal(TreeNode* root) {
    if (root == NULL) return;
    traversal(root->left);
    vec.push_back(root->val); // 将二叉搜索树转换为有序数组
    traversal(root->right);
}
public:
    int getMinimumDifference(TreeNode* root) {
        vec.clear();
        traversal(root);
        if (vec.size() < 2) return 0;
        int result = INT_MAX;
        for (int i = 1; i < vec.size(); i++) { // 统计有序数组的最小差值
            result = min(result, vec[i] - vec[i-1]);
        }
        return result;
    }
};

另附超快运行解法（来自leetcode）

class Solution {
TreeNode* pre = nullptr;
int res = INT_MAX;
public:
    int getMinimumDifference(TreeNode* root) {
        if(root == nullptr) return 0;

        getMinimumDifference(root->left);
        if(pre == nullptr) pre = root;
        else {res = min(res, root->val - pre->val); pre = root;}
        getMinimumDifference(root->right);

        return res;
    }
};

题目2、二叉搜索树中的众数

力扣题目链接(opens new window)

给定一个有相同值的二叉搜索树（BST），找出 BST 中的所有众数（出现频率最高的元素）。

假定 BST 有如下定义：

结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树

例如：

给定 BST [1,null,2,2],

501. 二叉搜索树中的众数

返回[2].

思路

因为本题给定的二叉树是二叉搜索树，所以二叉树的中序遍历就是有序的，遍历有序数组的元素出现频率，从头遍历，那么一定是相邻两个元素作比较，然后就把出现频率最高的元素输出就可以了。我们创建一个指针指向前一个节点，这样每次当前节点才能和前一个节点作比较。而且初始化的时候前一个节点为NULL，这样当前一个节点为NULL时候，我们就知道这是比较的第一个元素。

如果频率count 等于 maxCount（最大频率），当然要把这个元素加入到结果集中，频率大于最大频率的时候，不仅要更新最大频率，而且要清空结果集，因为结果集之前的元素都失效了。

代码实现

C++

class Solution {
private:
    int maxCount = 0; // 最大频率
    int count = 0; // 统计频率
    TreeNode* pre = NULL;
    vector<int> result;
    void searchBST(TreeNode* cur) {
        if (cur == NULL) return ;

        searchBST(cur->left);       // 左
                                    // 中
        if (pre == NULL) { // 第一个节点
            count = 1;
        } else if (pre->val == cur->val) { // 与前一个节点数值相同
            count++;
        } else { // 与前一个节点数值不同
            count = 1;
        }
        pre = cur; // 更新上一个节点

        if (count == maxCount) { // 如果和最大值相同，放进result中
            result.push_back(cur->val);
        }

        if (count > maxCount) { // 如果计数大于最大值频率
            maxCount = count;   // 更新最大频率
            result.clear();     // 很关键的一步，不要忘记清空result，之前result里的元素都失效了
            result.push_back(cur->val);
        }

        searchBST(cur->right);      // 右
        return ;
    }

public:
    vector<int> findMode(TreeNode* root) {
        count = 0;
        maxCount = 0;
        pre = NULL; // 记录前一个节点
        result.clear();

        searchBST(root);
        return result;
    }
};

题目3、二叉树的最近公共祖先

力扣题目链接(opens new window)

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

例如，给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]

236. 二叉树的最近公共祖先

示例 1: 输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1 输出: 3 解释: 节点 5 和节点 1 的最近公共祖先是节点 3。

示例 2: 输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4 输出: 5 解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。

思路

要想找到公共祖先，我们如果能自下而上地查找节点就好了，于是我们自然地想到回溯的思路，二叉树回溯的过程就是从下到上的。后序遍历（左右中）就是天然的回溯过程，可以根据左右子树的返回值，来处理中节点的逻辑。接下来我们还需要判断一个节点是节点q和节点p的公共祖先，在处理递归的逻辑上有很多细节，包括是否要处理返回值以及如何处理返回值，是否要搜索整棵二叉树等。

代码实现

C++

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == q || root == p || root == NULL) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (left != NULL && right != NULL) return root;

        if (left == NULL && right != NULL) return right;
        else if (left != NULL && right == NULL) return left;
        else  { //  (left == NULL && right == NULL)
            return NULL;
        }

    }
};

可以参考更优的题解（来自leetcode），这里不作注解

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == p || root == q || root == nullptr) {
            return root;
        }
        TreeNode* l = lowestCommonAncestor(root->left, p, q);
        TreeNode* r = lowestCommonAncestor(root->right, p, q);
        if (l != nullptr and r != nullptr) {
            return root;
        }
        if (l != nullptr) {
            return l;
        }
        return r;
    }
};

9月29日力扣每日一题

思路

对于在第k个人前的前k-1个人，如果这前k-1个人中有需要票数比第k个人的需要票数少的人，直接往count += tickets[i],否则加上第k个人的票数即可，对于第k个人后面的人，按相同思路再累加给count即可。

class Solution {
public:
    int timeRequiredToBuy(vector<int>& tickets, int k) {
        int count = 0; //用于记录时间
        int n = tickets.size();
        for (int i = 0; i < n; i++){
            if (i <= k){
                count += min(tickets[i], tickets[k]);
            }
            else{
                count += min(tickets[i], tickets[k] - 1);
            }
        }
        return count;
    }
};