Java中如何构建平衡二叉树

简介: Java中如何构建平衡二叉树



       平衡二叉树是一种二叉查找树,其中每个节点的左右两个子树的高度差不超过1。这种平衡性质确保了树的高度相对较小,使得在进行查找、插入和删除等操作时能够保持较高的性能。

定义:

平衡二叉树是一棵二叉排序树,或者为空,或者满足以下条件:

            1)左右子树高度差的绝对值不大于1;

            2)左右子树都是平衡二叉树。

平衡因子:

       左子树的高度减去右子树的高度,显然,在平衡二叉树中,每个结点的平衡因子的值为-1,0或1。

       在Java中,构造平衡二叉树通常涉及到 AVL 树(Adelson-Velsky and Landis tree)的实现。AVL 树是一种自平衡二叉搜索树,它通过保持每个节点的左右子树高度差不超过1来保持平衡。

插入操作的可能情况:

       LL型:新插入结点在A的左孩子(L)的左子树(L)中;

       LR型:新插入结点在A的左孩子(L)的右子树(R)中;

       RL型:新插入结点在A的右孩子(R)的左子树(L)中;

       RR型:新插入结点在A的右孩子(R)的右子树(R)中。

LL型

       右旋是指将一个节点的左子树变为其父节点,同时该节点成为其左子树的右子节点。右旋用于解决左子树过深的情况。

LR型

       右左旋是先对当前节点的右子树进行右旋,然后再对当前节点进行左旋。这用于解决右子树的左子树过深的情况。

RL型

       左右旋是先对当前节点的左子树进行左旋,然后再对当前节点进行右旋。这用于解决左子树的右子树过深的情况。

RR型

       左旋是指将一个节点的右子树变为其父节点,同时该节点成为其右子树的左子节点。左旋用于解决右子树过深的情况。

       以上都是基于左旋和右旋实现的,那么接下来我们构建左旋和右旋的方法以及插入数据对树结构所做操作:

对一组数据构建平衡二叉树:10、5、20、13、17、4

现在我们利用代码实现。

代码实现

       首先,需要创建一个 TreeNode 类,这个类除了包含节点值之外,还需要保存节点的高度:

public class TreeNode {
    int val;
    int height;
    TreeNode left;
    TreeNode right;
    public TreeNode(int val) {
        this.val = val;
        this.height = 1; // 初始高度为1
    }
}

       然后,创建一个 AVL 树的实现类,其中包括插入操作和平衡操作:

public class AVLTree {
  private TreeNode root;
    // 获取节点的高度
    private int height(TreeNode node) {
        if (node == null) {
            return 0;
        }
        return node.height;
    }
    // 获取平衡因子
    private int getBalance(TreeNode node) {
        if (node == null) {
            return 0;
        }
        return height(node.left) - height(node.right);
    }
    // 更新节点的高度
    private void updateHeight(TreeNode node) {
        if (node != null) {
            node.height = Math.max(height(node.left), height(node.right)) + 1;
        }
    }
    // 右旋转
    private TreeNode rightRotate(TreeNode y) {
      TreeNode x = y.left;
      TreeNode T2 = x.right;
        // 执行旋转
        x.right = y;
        y.left = T2;
        // 更新高度
        updateHeight(y);
        updateHeight(x);
        return x;
    }
    // 左旋转
    private TreeNode leftRotate(TreeNode x) {
      TreeNode y = x.right;
      TreeNode T2 = y.left;
        // 执行旋转
        y.left = x;
        x.right = T2;
        // 更新高度
        updateHeight(x);
        updateHeight(y);
        return y;
    }
    // 插入节点
    public TreeNode insert(TreeNode node, int val) {
        if (node == null) {
            return new TreeNode(val);
        }
        // 执行标准的BST插入
        if (val < node.val) {
            node.left = insert(node.left, val);
        } else if (val > node.val) {
            node.right = insert(node.right, val);
        } else {
            // 重复的值不被允许
            return node;
        }
        // 更新节点的高度
        updateHeight(node);
        // 获取平衡因子
        int balance = getBalance(node);
        // 左子树的左侧插入,需要右旋转
        if (balance > 1 && val < node.left.val) {
            return rightRotate(node);
        }
        // 右子树的右侧插入,需要左旋转
        if (balance < -1 && val > node.right.val) {
            return leftRotate(node);
        }
        // 左子树的右侧插入,需要先左旋转再右旋转
        if (balance > 1 && val > node.left.val) {
            node.left = leftRotate(node.left);
            return rightRotate(node);
        }
        // 右子树的左侧插入,需要先右旋转再左旋转
        if (balance < -1 && val < node.right.val) {
            node.right = rightRotate(node.right);
            return leftRotate(node);
        }
        return node;
    }
    // 插入新节点,外部调用该方法
    public TreeNode insert(int val) {
        root = insert(root, val);
        //返回根节点
        return root;
    }
    
}

       定义一个BinaryTreePrinter类,使用深度优先搜索来打印二叉树:

public class BinaryTreePrinter {
    public static void printTree(TreeNode root) {
        printTreeHelper(root, 0);
    }
    private static void printTreeHelper(TreeNode node, int depth) {
        if (node == null) {
            return;
        }
        // 打印右子树,使其显示在上方
        printTreeHelper(node.right, depth + 1);
        // 打印当前节点
        for (int i = 0; i < depth; i++) {
            System.out.print("    "); // 每层缩进四个空格
        }
        System.out.println(node.val);
        // 打印左子树,使其显示在下方
        printTreeHelper(node.left, depth + 1);
    }
}

       在测试类Test中创建了一个 AVLTree 实例,并调用其方法来操作 AVL 树:

public class Test {
  public static void main(String[] args) {
        AVLTree avlTree =new AVLTree();
        // 构造一棵平衡二叉树      
        avlTree.insert(5);
        avlTree.insert(10);
        avlTree.insert(20);
        avlTree.insert(13);
        avlTree.insert(17);
        TreeNode root=avlTree.insert(12);
        
        System.out.println("平衡二叉树高度:"+root.height);
        BinaryTreePrinter.printTree(root);
    }
}

       根节点为13,上下两部分分别是右子树和左子树,结果如下:

       与图示中构建结果相同。

总结

       AVL 的旋转问题看似复杂,但实际上如果你亲自用笔纸操作一下还是很好理解的。

相关文章
|
5月前
|
人工智能 算法 Java
Java与AI驱动区块链:构建智能合约与去中心化AI应用
区块链技术和人工智能的融合正在开创去中心化智能应用的新纪元。本文深入探讨如何使用Java构建AI驱动的区块链应用,涵盖智能合约开发、去中心化AI模型训练与推理、数据隐私保护以及通证经济激励等核心主题。我们将完整展示从区块链基础集成、智能合约编写、AI模型上链到去中心化应用(DApp)开发的全流程,为构建下一代可信、透明的智能去中心化系统提供完整技术方案。
389 3
|
6月前
|
人工智能 缓存 监控
使用LangChain4j构建Java AI智能体:让大模型学会使用工具
AI智能体是大模型技术的重要演进方向,它使模型能够主动使用工具、与环境交互,以完成复杂任务。本文详细介绍如何在Java应用中,借助LangChain4j框架构建一个具备工具使用能力的AI智能体。我们将创建一个能够进行数学计算和实时信息查询的智能体,涵盖工具定义、智能体组装、记忆管理以及Spring Boot集成等关键步骤,并展示如何通过简单的对话界面与智能体交互。
2000 1
|
6月前
|
安全 Java API
使用 Java 构建强大的 REST API 的四个基本技巧
本文结合探险领域案例,分享Java构建REST API的四大核心策略:统一资源命名、版本控制与自动化文档、安全防护及标准化异常处理,助力开发者打造易用、可维护、安全可靠的稳健API服务。
434 116
|
7月前
|
数据采集 搜索推荐 Java
Java 大视界 -- Java 大数据在智能教育虚拟学习环境构建与用户体验优化中的应用(221)
本文探讨 Java 大数据在智能教育虚拟学习环境中的应用,涵盖多源数据采集、个性化推荐、实时互动优化等核心技术,结合实际案例分析其在提升学习体验与教学质量中的成效,并展望未来发展方向与技术挑战。
|
6月前
|
人工智能 Java API
构建基于Java的AI智能体:使用LangChain4j与Spring AI实现RAG应用
当大模型需要处理私有、实时的数据时,检索增强生成(RAG)技术成为了核心解决方案。本文深入探讨如何在Java生态中构建具备RAG能力的AI智能体。我们将介绍新兴的Spring AI项目与成熟的LangChain4j框架,详细演示如何从零开始构建一个能够查询私有知识库的智能问答系统。内容涵盖文档加载与分块、向量数据库集成、语义检索以及与大模型的最终合成,并提供完整的代码实现,为Java开发者开启构建复杂AI智能体的大门。
3009 58
|
9月前
|
自然语言处理 前端开发 Java
JBoltAI 框架完整实操案例 在 Java 生态中快速构建大模型应用全流程实战指南
本案例基于JBoltAI框架,展示如何快速构建Java生态中的大模型应用——智能客服系统。系统面向电商平台,具备自动回答常见问题、意图识别、多轮对话理解及复杂问题转接人工等功能。采用Spring Boot+JBoltAI架构,集成向量数据库与大模型(如文心一言或通义千问)。内容涵盖需求分析、环境搭建、代码实现(知识库管理、核心服务、REST API)、前端界面开发及部署测试全流程,助你高效掌握大模型应用开发。
847 5
|
5月前
|
人工智能 缓存 自然语言处理
Java与多模态AI:构建支持文本、图像和音频的智能应用
随着大模型从单一文本处理向多模态能力演进,现代AI应用需要同时处理文本、图像、音频等多种信息形式。本文深入探讨如何在Java生态中构建支持多模态AI能力的智能应用。我们将完整展示集成视觉模型、语音模型和语言模型的实践方案,涵盖从文件预处理、多模态推理到结果融合的全流程,为Java开发者打开通往下一代多模态AI应用的大门。
478 41
|
5月前
|
设计模式 消息中间件 传感器
Java 设计模式之观察者模式:构建松耦合的事件响应系统
观察者模式是Java中常用的行为型设计模式,用于构建松耦合的事件响应系统。当一个对象状态改变时,所有依赖它的观察者将自动收到通知并更新。该模式通过抽象耦合实现发布-订阅机制,广泛应用于GUI事件处理、消息通知、数据监控等场景,具有良好的可扩展性和维护性。
456 8
|
5月前
|
机器学习/深度学习 人工智能 自然语言处理
Java与生成式AI:构建内容生成与创意辅助系统
生成式AI正在重塑内容创作、软件开发和创意设计的方式。本文深入探讨如何在Java生态中构建支持文本、图像、代码等多种生成任务的创意辅助系统。我们将完整展示集成大型生成模型(如GPT、Stable Diffusion)、处理生成任务队列、优化生成结果以及构建企业级生成式AI应用的全流程,为Java开发者提供构建下一代创意辅助系统的完整技术方案。
317 10
|
5月前
|
人工智能 监控 Java
Java与AI智能体:构建自主决策与工具调用的智能系统
随着AI智能体技术的快速发展,构建能够自主理解任务、制定计划并执行复杂操作的智能系统已成为新的技术前沿。本文深入探讨如何在Java生态中构建具备工具调用、记忆管理和自主决策能力的AI智能体系统。我们将完整展示从智能体架构设计、工具生态系统、记忆机制到多智能体协作的全流程,为Java开发者提供构建下一代自主智能系统的完整技术方案。
720 4