1 Star 0 Fork 0

yigemayun / 算法设计

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
matrixProduct.cpp 2.38 KB
一键复制 编辑 原始数据 按行查看 历史
yigemayun 提交于 2017-12-21 21:33 . add
#include <iostream>
#include <vector>
template<class T>
struct Matrix
{
Matrix(size_t r, size_t c)
{
Data.resize(c, std::vector<T>(r, 0));
}
void SetSubMatrix(const int r, const int c, const int n, const Matrix<T>& A, const Matrix<T>& B)
{
for(int _c=c; _c<n; ++_c)
{
for(int _r=r; _r<n; ++_r)
{
Data[_c][_r] = A.Data[_c][_r] + B.Data[_c][_r];
}
}
}
static Matrix<T> SquareMultiplyRecursive(Matrix<T>& A, Matrix<T>& B, int ar, int ac, int br, int bc, int n)
{
Matrix<T> C(n, n);
if(n == 1)
{
C.Data[0][0] = A.Data[ac][ar] * B.Data[bc][br];
}
else
{
C.SetSubMatrix(0, 0, n / 2,
SquareMultiplyRecursive(A, B, ar, ac, br, bc, n / 2),
SquareMultiplyRecursive(A, B, ar, ac + (n / 2), br + (n / 2), bc, n / 2));
C.SetSubMatrix(0, n / 2, n / 2,
SquareMultiplyRecursive(A, B, ar, ac, br, bc + (n / 2), n / 2),
SquareMultiplyRecursive(A, B, ar, ac + (n / 2), br + (n / 2), bc + (n / 2), n / 2));
C.SetSubMatrix(n / 2, 0, n / 2,
SquareMultiplyRecursive(A, B, ar + (n / 2), ac, br, bc, n / 2),
SquareMultiplyRecursive(A, B, ar + (n / 2), ac + (n / 2), br + (n / 2), bc, n / 2));
C.SetSubMatrix(n / 2, n / 2, n / 2,
SquareMultiplyRecursive(A, B, ar + (n / 2), ac, br, bc + (n / 2), n / 2),
SquareMultiplyRecursive(A, B, ar + (n / 2), ac + (n / 2), br + (n / 2), bc + (n / 2), n / 2));
}
return C;
}
void Print()
{
for(int c=0; c<Data.size(); ++c)
{
for(int r=0; r<Data[0].size(); ++r)
{
std::cout << Data[c][r] << " ";
}
std::cout << "\n";
}
std::cout << "\n";
}
std::vector<std::vector<T> > Data;
};
int main()
{
Matrix<int> A(2, 2);
Matrix<int> B(2, 2);
A.Data[0][0] = 2;
A.Data[0][1] = 1;
A.Data[1][0] = 1;
A.Data[1][1] = 2;
B.Data[0][0] = 2;
B.Data[0][1] = 1;
B.Data[1][0] = 1;
B.Data[1][1] = 2;
A.Print();
B.Print();
Matrix<int> C(Matrix<int>::SquareMultiplyRecursive(A, B, 0, 0, 0, 0, 2));
C.Print();
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yigemayun/SuanFaSheJi.git
git@gitee.com:yigemayun/SuanFaSheJi.git
yigemayun
SuanFaSheJi
算法设计
master

搜索帮助

344bd9b3 5694891 D2dac590 5694891