#include <iostream>
using namespace std;
//将矩阵a和b相加得到c
template <class T>
void matrixAdd(T **a, T **b, T **c, int numberOfRows, int numberOfColumns)
{
for (int i = 0; i < numberOfRows; i++)
for (int j = 0; j < numberOfColumns; j++)
c[i][j] = a[i][j] + b[i][j];
}
//两个n*n矩阵的乘法;将矩阵a和b相乘得到c
template <class T>
void squareMatrixMultiply(T **a, T **b, T **c, int n)
{
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
{
T sum = 0;
for (int k = 0; k < n; k++)
{
sum += a[i][k] * b[k][j];
}
c[i][j] = sum;
}
}
}
//m*n矩阵与n*p矩阵的乘法
template <class T>
void matrixMultiply(T **a, T **b, T **c, int n,int p)
{
for (int i = 0; i < m; i++)
{
for (int j = 0; j < p; j++)
{
for (int k = 0; k < n; k++)
{
sum += a[i][k] * b[k][j];
}
a[i][j] = sum;
}
}
}
int main()
{
int arr1[3] = { 1, 2, 3 }; //定义一维数组
int arr2[3] = { 4, 5, 6 };
int arr3[3] = { 7, 8, 9 };
int arr4[3] = { 10, 11, 12 };
int *a[3] = { arr1, arr2, arr3 }; //指针数组
int *b[3] = { arr1, arr2, arr3 };
int *c[4] = { arr1, arr2, arr3, arr4 };
int *result1[3];
int *result2[3];
int *result3[4];
int **f = a;
matrixAdd(a, b, result1, 3, 3);
squareMatrixMultiply(a, b, result2, 3);
matrixMultiply(a, b, result3, 3, 4);
return 0;
}
C++ 矩阵加法与乘法的实现
最新推荐文章于 2024-08-02 11:17:36 发布