【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 大于零时重复上述步骤。

相关推荐
算AI15 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
懒羊羊大王&16 小时前
模版进阶(沉淀中)
c++
owde17 小时前
顺序容器 -list双向链表
数据结构·c++·链表·list
GalaxyPokemon17 小时前
Muduo网络库实现 [九] - EventLoopThread模块
linux·服务器·c++
W_chuanqi17 小时前
安装 Microsoft Visual C++ Build Tools
开发语言·c++·microsoft
hyshhhh17 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
tadus_zeng18 小时前
Windows C++ 排查死锁
c++·windows
EverestVIP18 小时前
VS中动态库(外部库)导出与使用
开发语言·c++·windows
杉之18 小时前
选择排序笔记
java·算法·排序算法
烂蜻蜓18 小时前
C 语言中的递归:概念、应用与实例解析
c语言·数据结构·算法