最优二叉搜索树 - Java实现（递归，带备忘录递归，动态规划）

最新推荐文章于 2022-10-13 19:47:12 发布

InDeepWell

最新推荐文章于 2022-10-13 19:47:12 发布

阅读量1.3k

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/JunyuLee/article/details/105959148

版权

本文探讨如何使用动态规划方法构建期望搜索代价最小的最优二叉搜索树。通过对比矩阵链连乘问题，阐述从length = 1逐步到length = n的动态规划解法。详细讲解了遍历过程，并给出了一组数据测试。

摘要由CSDN通过智能技术生成

对于一个给定的概率集合，我们希望构造一棵期望搜索代价最小的二叉搜索树，我们称之为最优二叉搜索树

最近复习了一下算法导论的动态规划，发现最优二叉搜索树的实现方法与矩阵链连乘问题极其相似。
根据动态规划bottom -> top的思想，我们应该从length = 1的情况逐渐求解到length = n的情况，对于每一个i = length，我们从起始位置 j = 1开始，加上length - 1作为当前序列的end index,遍历 j = 1 -> n - i + 1（因为每一次长度增长，对应的可遍历的序列数量就会 - 1，打个比方，如果二叉搜索树序列为<1,4,5,6>，第一次遍历时会分别求出cost[1,1],cost[2,2],cost[3,3]和cost[4,4]，第二次遍历仍然从index = 1开始，但是end index = 1 + 2 - 1 = 1，也就是第二次遍历会求出cost[1,2],cost[2,3]和cost[3,4]…按照这种遍历方式，最终求出cost[1,n]，n是二叉搜索树序列长度，同时使用一个n * n数组root记录从i -> j的节点的父节点）

public class OptimalBinarySearchTree {
   
    private static final double[] intlNodes = {
   0, 0.15, 0.1, 0.05, 0.1, 0.2};
    private static final double[] leafNodes = {
   0.05, 0.1, 0.05, 0.05, 0.05, 0.1};
    private static final int length = intlNodes.length;

    /**
     * This is a brutal solution
     * Top -> Bottom
     * @param i 关键字搜索起点
     * @param j 关键字搜索终点
     * @param realNodes 关键字数组
     * @param fakeNodes 伪关键字数组
     * @return  最小搜索期望
     */
    private static double solution(int i, int j, double[] realNodes, double[] fakeNodes){
   
        //平凡case,子树不再包含关键字，而只有一个伪关键字，返回当前伪关键字
        if(j == i - 1){
   
            return fakeNodes[j];
        }
        //否则计算从i -> j的关键字的最小搜索期望
        double minExpect = Double.MAX_VALUE;
        double weight = 0;
        //根据w[i,j] = 包含了i -> j的关键字节点和伪关键字节点的概率和
        weight += fakeNodes[i-1];
        for