首先构造一个结构体表示矩阵(当然可以构造成类,然后将所有方法集成到该类中,这里简化操作):
const int N = 3;
struct Matrix{
int row, col;
int matrix[N][N] = {0};
Matrix(){}
Matrix(int r, int c): row(r), col(c){}
};
对于矩阵的转置,其实很简单实现:
Matrix Reverse(Matrix x){
Matrix ans = Matrix(x.col, x.row);
for(int i = 0;i < ans.row;i++){
for(int j = 0;j < ans.col;j++){
ans.matrix[i][j] = x.matrix[j][i];
}
}
return ans;
}
只需要行列互换即可,而矩阵的加减法是类似的,因此这里只给出加法运算:
Matrix Add(Matrix x, Matrix y){
Matrix ans = Matrix(x.row, x.col);
for(int i = 0;i < ans.row;i++){
for(int j = 0;j < ans.col;j++){
ans.matrix[i][j] = x.matrix[i][j] + y.matrix[i][j];
}
}
return ans;
}
矩阵的乘法根据线代中的知识可以得到:
Matrix Multiply(Matrix x, Matrix y){
Matrix ans = Matrix(x.row, y.col);
for(int i = 0;i < ans.row;i++){
for(int j = 0;j < ans.col;j++){
for(int k = 0;k < x.col;k++){
ans.matrix[i][j] += x.matrix[i][k] * y.matrix[k][j];
}
}
}
return ans;
}
矩阵的幂乘简言之就是n个一样的矩阵相乘,我们当然可以循环相乘得到答案,但是这样的效率就会很低,因此这里给出了一种快速幂的方法:
Matrix QuickPower(Matrix x, int n){
Matrix ans = Matrix(x.row, x.col);
for(int i = 0;i < ans.row;i++){ //先将ans初始化为单位矩阵
for(int j = 0;j < ans.col;j++){
if(i == j) ans.matrix[i][j] = 1;
else ans.matrix[i][j] = 0;
}
}
while(n != 0){
if(n % 2 == 1) ans = Multiply(ans, x);
n /= 2;
x = Multiply(x, x);
}
return ans;
}