题目描述
一个n*m矩阵由n行m列共n*m个数排列而成。两个矩阵A和B可以相乘当且仅当A的列数等于B的行数。一个N*M的矩阵乘以一个M*P的矩阵等于一个N*P的矩阵,运算量为nmp。
矩阵乘法满足结合律,A*B*C可以表示成(A*B)*C或者是A*(B*C),两者的运算量却不同。例如当A=2*3 B=3*4 C=4*5时,(A*B)*C=64而A*(B*C)=90。显然第一种顺序节省运算量。
现在给出N个矩阵,并输入N+1个数,第i个矩阵是a[i-1]*a[i]
输入
第一行n(n<=100)
第二行n+1个数
输出
最优的运算量
样例输入
3 2 3 4 5
样例输出
64
分析:
其实该题使用了动态规划来选出最优子结构,并且使用了以下等式:
先初始化m数组和s数组,这里使用了C++的函数memset():
void* memset(void* s, int c, size_t n);
参数解释:
- `s`:指向要填充的内存区域的指针。
- `c`:要设置的字符值(实际上是将其转换为对应的ASCII码或字节值)。
- `n`:要填充的字节数。
`memset`函数将`s`指向的内存区域的前`n`个字节用`c`指定的值进行填充。返回值是原始的`s`指针。
并且m矩阵里面其实填充的是矩阵的右上角的部分,如可以看作下图这样:
代码:
//矩阵连乘
#include<iostream>
#include<cstring>
using namespace std;
const int size = 101;
int p[size];
int m[size][size], s[size][size];
int n;void matrixchain()
{int i, r, j, k;memset(m, 0, sizeof(m));memset(s, 0, sizeof(s));//初始化数组for (r = 2; r <= n; r++)//矩阵连乘的规模为r {for (i = 1; i <= n - r + 1; i++){j = i + r - 1;m[i][j] = m[i + 1][j] + p[i - 1] * p[i] * p[j];//对m[][]开始赋值s[i][j] = i;//s[][]存储各子问题的决策点for (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;s[i][j] = k;}}}}
}int main()
{cin >> n;int i, j;for (i = 0; i <= n; i++)cin >> p[i];matrixchain();cout << m[1][n] << endl;return 0;
}
参考博文:动态规划之——矩阵连乘(全网最详细博文,看这一篇就够了!)_矩阵连乘问题-CSDN博客