矩阵连乘问题

《计算机算法设计与分析》

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2023/1/8 21:48
# @Author  : Lili
# @File    : 矩阵连乘.py
# @Description : 求解，计算n个矩阵的连乘积的最优次序，使得数乘次数最少
# 基本思想：将带求解问题分解成若干子问题，先求解子问题，再结合这些子问题的解得到原问题的解。
# 而这些子问题往往是不相互独立的，因此通过表记录与查表的方法来避免重复计算，从而达到多项式时间算法。

# 计算步骤：
#  找出最优解的性质，并刻画其结构特征
#  递归地定义最优值
#  以自底向上的方式计算最优值
#  根据计算最优值时得到的信息，构造最优解

# 基本要素：
#  最优子结构性质、重叠子问题性质
import array


def matrix_chain(p: list):
    """
    计算矩阵连乘积最优计算次序的动态规划算法

    :param p: 矩阵的行列，重复部分省略，均满足矩阵的乘法
    :return: m[0][-1], s：最少数乘次数，计算次序
    """
    n = len(p) - 1  # p所代表的矩阵的个数
    m = []  # 子问题最优解
    s = [[0 for j in range(0, n)]
         for i in range(0, n)]  # 记录计算次序
    for i in range(0, n):
        m.append(array.array('i', [0 for j in range(0, n)]))
    for r in range(1, n):
        for i in range(0, n-r):
            j = i+r
            # 递归公式，但改变计算次序，来充分利用已经计算过的子问题的解
            m[i][j] = m[i+1][j] + p[i]*p[i+1]*p[j+1]
            s[i][j] = i
            for k in range(i+1, j):
                t = m[i][k] + m[k+1][j] + p[i]*p[k+1]*p[j+1]
                if t < m[i][j]:
                    m[i][j], s[i][j] = t, k
    return m[0][-1], s[0]


def memoized_matrix_chain(p: list):
    """
    备忘录方法解矩阵连乘，动态规划法的变形，在递归的同时记表查表，避免重复计算子问题

    :param p: 矩阵的行列，重复部分省略，均满足矩阵的乘法
    :return: m[0][-1], s：最少数乘次数，计算次序
    """
    n = len(p)-1
    m = []  # 子问题最优解
    s = [[0 for j in range(0, n)]
         for i in range(0, n)]  # 记录计算次序
    for i in range(0, n):
        m.append(array.array('i', [0 for j in range(0, n)]))

    def lookup_chain(i: int, j: int):
        if m[i][j] > 0:
            return m[i][j]
        if i == j:
            return 0
        u = lookup_chain(i, i) + lookup_chain(i + 1, j) + p[i] * p[i+1] * p[j+1]
        s[i][j] = i
        for k in range(i + 1, j):
            t = lookup_chain(i, k) + lookup_chain(k + 1, j) + p[i] * p[k+1] * p[j+1]
            if t < u:
                u, s[i][j] = t, k
        m[i][j] = u
        return u

    result = lookup_chain(0, n-1)
    return result, s[0]


p = [30, 35, 15, 5, 10, 20, 25]
m, s = memoized_matrix_chain(p)
# m, s = matrix_chain(p)
print("最少数乘次数：", m)
# 输出乘法顺序
bra = [""] * len(p)
brackets = set()
for i in reversed(range(1, len(p)-1)):
    begin, mid, end = 0, s[i]+1, i+1
    if mid - begin > 1:
        brackets.add((begin, mid))
    if end - mid > 1:
        brackets.add((mid, end))
for bracket in brackets:
    bra[bracket[0]] += '('
    bra[bracket[1]] += ')'
arr = [bra[i]+" A"+str(i)+" " for i in range(0, len(p)-1)]
arr.append(bra[-1])
print("".join(arr))