【c++&leetcode】1382. Balance a Binary Search Tree

问题入口

DSW (DAY, STOUT & WARREN) ALGORITHM

时间复杂度O(n)

cpp 复制代码
class Solution {
public:
    int makeVine(TreeNode* grand, int cnt = 0)
    {
        auto n = grand->right;
        while (n != nullptr)
        {
            if(n->left != nullptr)
            {
                auto old_n = n;
                n = n->left;
                old_n->left = n->right;
                n->right = old_n;
                grand->right = n;
            }
            else{
                cnt++;
                grand = n;
                n = n->right;
            }
        }
        return cnt;
        
    }
    void compress(TreeNode *grand, int m) {
        auto n = grand->right;
        while (m-- > 0)
        {
            auto old_n = n;
            n = n->right;
            grand->right = n;
            old_n->right = n->left;
            n->left = old_n;
            grand = n;
            n = n->right;
        }
        
    }
    TreeNode* balanceBST(TreeNode *root) {
        TreeNode grand;
        grand.right = root;
        auto cnt = makeVine(&grand);
        int m = pow(2, int(log2(cnt + 1))) - 1;
        compress(&grand, cnt - m);
        for (m = m / 2; m > 0; m /= 2)
            compress(&grand, m);
        return grand.right;
    }
};
  1. 将最初的树转变为藤蔓。 通过进行右旋转,我们将树展平为"链表",其中头是以前最左边的节点,尾部是以前最右边的节点。

  2. 将树转换为藤蔓后,计算 节点总数(cnt)。

  1. 计算最接近的完美平衡树的高度:h = log2(cnt + 1)。

  2. 计算最接近的完美平衡树中的节点数:m = pow(2, h) - 1。

  3. 向左旋转 cnt - m 个节点以掩盖多余的节点。

  1. 左旋转 m/2 个节点。

  2. 将 m 除以 2,并在 m / 2 大于零时重复上述步骤。

相关推荐
喵先生!3 小时前
C++中的vector和list的区别与适用场景
开发语言·c++
xMathematics4 小时前
计算机图形学实践:结合Qt和OpenGL实现绘制彩色三角形
开发语言·c++·qt·计算机图形学·cmake·opengl
ShiinaMashirol4 小时前
代码随想录打卡|Day27(合并区间、单调递增的数字、监控二叉树)
java·算法
yuanManGan6 小时前
C++入门小馆: 深入了解STLlist
开发语言·c++
梁下轻语的秋缘6 小时前
每日c/c++题 备战蓝桥杯(P1049 [NOIP 2001 普及组] 装箱问题)
c语言·c++·学习·蓝桥杯
逐光沧海6 小时前
STL常用算法——C++
开发语言·c++
wuqingshun3141596 小时前
蓝桥杯 5. 交换瓶子
数据结构·c++·算法·职场和发展·蓝桥杯
Demons_kirit6 小时前
Leetcode 2845 题解
算法·leetcode·职场和发展
球求了7 小时前
C++:继承机制详解
开发语言·c++·学习
adam_life7 小时前
http://noi.openjudge.cn/——2.5基本算法之搜索——200:Solitaire
算法·宽搜·布局唯一码