首页 > 代码库 > 算法设计与分析之动态规划

算法设计与分析之动态规划

今天刚刚学了动态规划,把书上的代码敲了一下,在此留下一笔。动态规划与分治法类似,其基本思想也是将待求解问题分解成若干个子问题,先求解子问题,然后从这些子问题的解得到原问题的解。在子问题的求解过程中,有很多子问题被重复计算了。于是我们可以用一个表来记录所有已解决的子问题的的答案。不管该问题以后是否被用到,只要它被计算过,就将其结果填入表中。这就是动态规划的基本思想。

一、动态规划的设计步骤:

(1)找出最优解的性质,并刻画其结构特征;

(2)递归地定义最优值;

(3)以自顶向下的方式计算出最优值;

(4)根据计算最优值时得到的信息,构造最优解。

下面使用动态规划来解决的是矩阵连乘问题。

 问题描述:给定n个矩阵:A1,A2,...,An,其中Ai与Ai+1是可乘的,i=1,2...,n-1。确定计算矩阵连乘积的计算次序,使得依此次序计算矩阵连乘积需要的数乘次数最少。输入数据为矩阵个数和每个矩阵规模,输出结果为计算矩阵连乘积的计算次序和最少数乘次数。

第一步分析最优解的结构:

  若将对应m[i][j]的断开位置k记为s[i][j],在计算出最优值m[i][j]后,可递归地由s[i][j]构造出相应的最优解。s[i][j]中的数表明,计算矩阵链A[i:j]的最佳方式应在矩阵Ak和Ak+1之间断开,即最优的加括号方式应为(A[i:k])(A[k+1:j)。因此,从s[1][n]记录的信息可知计算A[1:n]的最优加括号方式为(A[1:s[1][n]])(A[s[1][n]+1:n]),进一步递推,A[1:s[1][n]]的最优加括号方式为(A[1:s[1][s[1][n]]])(A[s[1][s[1][n]]+1:s[1][s[1][n]]])。同理可以确定A[s[1][n]+1:n]的最优加括号方式在s[s[1][n]+1][n]处断开...照此递推下去,最终可以确定A[1:n]的最优完全加括号方式,及构造出问题的一个最优解。

 

第二步、建立递归关系

 

 设计算A[i:j],1≤i≤j≤n,所需要的最少数乘次数m[i,j],则原问题的最优值为m[1,n]。

      当i=j时,A[i:j]=Ai,因此,m[i][i]=0,i=1,2,…,n
      当i<j时,若A[i:j]的最优次序在Ak和Ak+1之间断开,i<=k<j,则:m[i][j]=m[i][k]+m[k+1][j]+pi-1pkpj。由于在计算是并不知道断开点k的位置,所以k还未定。不过k的位置只有j-i个可能。因此,k是这j-i个位置使计算量达到最小的那个位置。

      综上,有递推关系如下:

      技术分享    

 

 

 

第三步、计算最优解

 

 

 

最后代码如下:

package algo.chapter3;

import java.io.Serializable;

/**
 * 矩阵相乘
 * @author Administrator
 *
 */
public class MatrixPlus {
    

    public static void matrixChain(int[] p,int [][]m,int [][]s)
    {
        int n = p.length-1;
        for(int i=1;i<=n;i++)
            m[i][i] = 0;
        for(int r=2;r<=n;r++)
            //对角线上循环的次数
            for(int i=1;i<=n-r+1;i++)
            {
                int j = i+r-1;
                m[i][j] = m[i+1][j] + p[i-1]*p[i]*p[j];
                s[i][j] = i;
                for(int k=i+1;k<j;k++)
                {
                    int t = m[i][k] + m[k+1][j] + p[i-1]*p[k]*p[j];
                    if(t < m[i][j])
                    {
                        m[i][j] = t;
                        //递归式,记录k的位置
                        s[i][j] = k;
                    }
                }
            }
    }
    
    

    public static void traceback(int [][]s,int i,int j)
    {
        if(i == j)
            return;
        traceback(s,i,s[i][j]);
        traceback(s,s[i][j]+1,j);
        System.out.println("Multiply A"+i+"."+s[i][j]+
                "and A"+(s[i][j]+1)+"."+j);
    }
    
    //打印加括号的最优解方案
    public static void optimalParens(int [][]s,int i,int j)
    {
        if(i==j)
            System.out.print("(A"+i);
        else
        {
            optimalParens(s,i,s[i][j]);
            optimalParens(s,s[i][j]+1,j);
            System.out.print(")");
        }
    }
}

 

测试代码:

package algo.chapter3.test;

import algo.chapter3.MatrixPlus;

public class MatrixTest {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        int [] p = {39,45,35,15,23,45,67,10,26,37};
        int n = p.length;
        //{{12,3,45,56},{2,4,6,23},{9,87,34,23},{23,45,56,56},{23,3,6,8}}
        int [][] m = new int[n][n];
        int [][] s = new int[n][n];
        MatrixPlus.matrixChain(p, m, s);
        System.out.println("该矩阵阶乘子问题数乘的个数为:");
        for(int i=1;i<m.length;i++)
        {
            for(int j=1;j<m.length;j++)
            {    
                if(i>j)
                {
                    System.out.print("----"+"\t");
                }
                else
                {
                    System.out.print(m[i][j]+"\t");
                }
            }
            System.out.println();
        }
        System.out.println();
        System.out.println("该矩阵阶乘子问题数乘的个数为:");
        for(int i=1;i<s.length;i++)
        {
            for(int j=1;j<s.length;j++)
            {    
                if(i>j)
                {
                    System.out.print("----"+"\t");
                }
                else
                {
                    System.out.print(s[i][j]+"\t");
                }
            }
            System.out.println();
        }
        
        System.out.println();
        System.out.println("该矩阵阶乘的最优解为:");
        MatrixPlus.traceback(s, 1, n-1);
        MatrixPlus.optimalParens(s, 1, n-1);
    }

}

最后的运行结果如下图:

技术分享

 

技术分享

 参考:http://blog.csdn.net/liufeng_king/article/details/8497607

算法设计与分析之动态规划