【蓝桥杯】算法提高 矩阵乘法(C++)

【蓝桥杯】算法提高 矩阵乘法(C++)

问题描述

题目链接:矩阵乘法.
问题描述:
  有n个矩阵,大小分别为a0a1, a1a2, a2a3, …, a[n-1]a[n],现要将它们依次相乘,只能使用结合率,求最少需要多少次运算。
  两个大小分别为p
q和q
r的矩阵相乘时的运算次数计为pqr。
输入格式:
  输入的第一行包含一个整数n,表示矩阵的个数。
  第二行包含n+1个数,表示给定的矩阵。
输出格式:
  输出一个整数,表示最少的运算次数。
样例输入:

3
1 10 5 20

样例输出:

150

数据规模和约定:
  1<=n<=1000, 1<=ai<=10000。

解题思路

借鉴:【算法导论】动态规划之“矩阵链乘法”问题.
1.用dp[i][j]代表矩阵i 连续相乘到 j 的最小运算数
2.状态转移方程: dp[i][j] = min(dp[i][j],dp[i][k] + dp[k + 1][j] + a[i - 1]*a[k]a[j]
其中 k 代表将{i···j}原矩阵链分割成 {i··k}
{k+1···j}两小矩阵相乘

具体代码

#include<bits/stdc++.h>
using namespace std;
long long n,a[1001],dp[
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是使用C语言实现的Strassen标准矩阵乘法算法的代码: ``` #include <stdio.h> #define N 4 // 矩阵大小 #define THRESHOLD 2 // 阈值 // 矩阵加法 void add_matrix(int a[][N], int b[][N], int c[][N]) { int i, j; for (i = 0; i < N; i++) { for (j = 0; j < N; j++) { c[i][j] = a[i][j] + b[i][j]; } } } // 矩阵减法 void sub_matrix(int a[][N], int b[][N], int c[][N]) { int i, j; for (i = 0; i < N; i++) { for (j = 0; j < N; j++) { c[i][j] = a[i][j] - b[i][j]; } } } // Strassen矩阵乘法 void strassen_mul(int a[][N], int b[][N], int c[][N]) { // 达到阈值,使用标准矩阵乘法 if (N <= THRESHOLD) { int i, j, k; for (i = 0; i < N; i++) { for (j = 0; j < N; j++) { c[i][j] = 0; for (k = 0; k < N; k++) { c[i][j] += a[i][k] * b[k][j]; } } } return; } // 处理矩阵的大小并向上取整 int size = N / 2; if (N % 2 != 0) { size += 1; } int A[size][size], B[size][size], C[size][size], D[size][size]; int E[size][size], F[size][size], G[size][size], H[size][size]; int P1[size][size], P2[size][size], P3[size][size], P4[size][size], P5[size][size], P6[size][size], P7[size][size]; int tmp1[size][size], tmp2[size][size]; // 拆分矩阵 int i, j; for (i = 0; i < size; i++) { for (j = 0; j < size; j++) { A[i][j] = a[i][j]; B[i][j] = a[i][j + size]; C[i][j] = a[i + size][j]; D[i][j] = a[i + size][j + size]; E[i][j] = b[i][j]; F[i][j] = b[i][j + size]; G[i][j] = b[i + size][j]; H[i][j] = b[i + size][j + size]; } } // 计算P1到P7 sub_matrix(F, H, tmp1); strassen_mul(A, tmp1, P1); add_matrix(A, B, tmp1); strassen_mul(tmp1, H, P2); add_matrix(C, D, tmp1); strassen_mul(tmp1, E, P3); sub_matrix(G, E, tmp1); strassen_mul(D, tmp1, P4); add_matrix(A, D, tmp1); add_matrix(E, H, tmp2); strassen_mul(tmp1, tmp2, P5); sub_matrix(B, D, tmp1); add_matrix(G, H, tmp2); strassen_mul(tmp1, tmp2, P6); sub_matrix(A, C, tmp1); add_matrix(E, F, tmp2); strassen_mul(tmp1, tmp2, P7); // 计算结果矩阵 add_matrix(P5, P4, tmp1); sub_matrix(tmp1, P2, tmp2); add_matrix(tmp2, P6, c[0]); add_matrix(P1, P2, c[1]); add_matrix(P3, P4, c[2]); add_matrix(P5, P1, tmp1); sub_matrix(tmp1, P3, tmp2); sub_matrix(tmp2, P7, c[3]); } int main() { int a[N][N] = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}}; int b[N][N] = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}}; int c[N][N]; int i, j; strassen_mul(a, b, c); printf("Result:\n"); for (i = 0; i < N; i++) { for (j = 0; j < N; j++) { printf("%d ", c[i][j]); } printf("\n"); } return 0; } ``` 本代码中定义了THRESHOLD变量,当矩阵大小小于等于阈值时,使用标准矩阵乘法算法计算。简单起见,本代码中矩阵大小固定为4 * 4,可以根据需要修改。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值