线性代数矩阵相乘

代码优化分析

冗余变量问题

初始代码中提前定义了 m=2,n=3; f=3,p=2;,但后续通过 cin 重新输入覆盖了这些值,导致初始赋值完全无效。建议删除这些冗余的初始化语句。

内层循环改进

原代码使用双变量 jb 进行内层循环:

cpp 复制代码
for (int j = 0,b=0; j < n&&b<f; ++j,++b)

由于 n == f,只需使用单变量 j 进行遍历即可,双变量写法降低了代码可读性。

重复输出问题

原代码在计算过程中打印了一次结果,随后又完整遍历输出矩阵,属于多余操作。优化后仅在最后输出一次结果矩阵。

变量命名规范
  • f 代表第二个矩阵的行数,建议改为 n2 以提高可读性。
  • ab 等变量名不够直观,建议改为 colk 以符合矩阵编程习惯。
输出格式美化

原代码输出矩阵时未进行分行对齐,影响观感。优化后使用制表符 \t 对齐输出。

优化后代码

cpp 复制代码
#include <iostream>
#include <vector>

int main() {
    int m, n, p;
    std::cout << "请输入矩阵A的行数 列数:";
    std::cin >> m >> n;

    int n2, p2;
    std::cout << "请输入矩阵B的行数 列数:";
    std::cin >> n2 >> p2;

    if (n != n2) {
        std::cout << "矩阵行列不匹配,无法相乘!" << std::endl;
        return 1;
    }

    std::vector<std::vector<int>> A(m, std::vector<int>(n, 0));
    std::vector<std::vector<int>> B(n2, std::vector<int>(p2, 0));
    std::vector<std::vector<int>> C(m, std::vector<int>(p2, 0));

    std::cout << "\n请输入矩阵A(" << m << "行" << n << "列)元素:" << std::endl;
    for (int i = 0; i < m; ++i) {
        for (int j = 0; j < n; ++j) {
            std::cin >> A[i][j];
        }
    }

    std::cout << "\n请输入矩阵B(" << n2 << "行" << p2 << "列)元素:" << std::endl;
    for (int i = 0; i < n2; ++i) {
        for (int j = 0; j < p2; ++j) {
            std::cin >> B[i][j];
        }
    }

    for (int i = 0; i < m; ++i) {
        for (int j = 0; j < p2; ++j) {
            int sum = 0;
            for (int k = 0; k < n; ++k) {
                sum += A[i][k] * B[k][j];
            }
            C[i][j] = sum;
        }
    }

    std::cout << "\n矩阵 A * B 的结果:" << std::endl;
    for (int i = 0; i < m; ++i) {
        for (int j = 0; j < p2; ++j) {
            std::cout << C[i][j] << "\t";
        }
        std::cout << std::endl;
    }

    return 0;
}

核心逻辑对应数学规则

矩阵乘法的三层循环对应以下数学公式: C_{i,j} = \\sum_{k=0}\^{n-1} A_{i,k} \\times B_{k,j}

  • 外层循环 i:结果矩阵的行,对应左矩阵 A 的行。
  • 中层循环 j:结果矩阵的列,对应右矩阵 B 的列。
  • 内层循环 k:遍历 A 的行和 B 的列,进行对位相乘并累加。

测试示例

输入矩阵 A(2行3列):

复制代码
1 0 2
-1 3 1

输入矩阵 B(3行2列):

复制代码
4 1
2 0
-3 5

输出结果:

复制代码
-2      11
-1      4

与手工计算结果一致。

补充知识点

  • 矩阵乘法不满足交换律,A*BB*A 结果通常不同。
  • 二维 vector 适用于动态矩阵,但在工程中大规模运算建议使用数组或专用矩阵库。
  • 避免使用双变量循环写法,改用单变量 k 进行遍历。
相关推荐
wabs6665 小时前
关于动态规划【力扣1143.最长公共子序列的思考】
算法·leetcode·动态规划
剑挑星河月6 小时前
54.螺旋矩阵
java·算法·leetcode·矩阵
Robot_Nav6 小时前
MPPI 局部规划器实验设计讲解
人工智能·算法·mppi
mingo_敏7 小时前
Mean-Teacher 均值教师自训练框架详解
算法·均值算法
星空露珠7 小时前
迷你世界UGc3.0脚本Wiki[剧情动画模块管理接口 Timeline]
开发语言·数据结构·算法·游戏·lua
笨笨没好名字7 小时前
Leetcode刷题python3版第一周(下)
linux·算法·leetcode
手写码匠7 小时前
手写 LLM 安全护栏:从内容审核到越狱防御的完整实现
人工智能·深度学习·算法·aigc
luj_17687 小时前
草酸与烟酸对消化及糖代谢的影响解析
服务器·c语言·开发语言·经验分享·算法
青风978 小时前
16-ADAPTRACK:基于自适应阈值的多目标跟踪匹配算法
人工智能·算法·目标跟踪
汤姆yu8 小时前
macOS系统下Aider完整安装、配置与实战使用教程
大数据·人工智能·算法·macos·github·copilot