92 lines
2.5 KiB
C
92 lines
2.5 KiB
C
|
//
|
||
|
|
// Created by 顾涵彬 on 2019-08-29.
|
||
|
|
//
|
||
|
|
|
||
|
|
#ifndef MATRIX_SQUAREMATRIX_H
|
||
|
|
#define MATRIX_SQUAREMATRIX_H
|
||
|
|
#include "Matrix.h"
|
||
|
|
namespace Ctain {
|
||
|
|
#define Matrix Matrix<_Scalar>
|
||
|
|
template<typename _Scalar>
|
||
|
|
class SMatrix: public Matrix{
|
||
|
|
public:
|
||
|
|
SMatrix(int D) : Matrix(D, D) {}
|
||
|
|
SMatrix() : Matrix(0, 0) {}
|
||
|
|
SMatrix(_Scalar _data[], int D) :
|
||
|
|
Matrix(_data, D, D) {}
|
||
|
|
SMatrix(_Scalar **_data, int D) :
|
||
|
|
Matrix(_data, D, D) {}
|
||
|
|
SMatrix(Matrix m) :
|
||
|
|
Matrix(m) {}
|
||
|
|
// void operator =(const Matrix &m){
|
||
|
|
// }
|
||
|
|
_Scalar determinant();
|
||
|
|
_Scalar M(int m, int n);
|
||
|
|
SMatrix<_Scalar> inverse() {
|
||
|
|
SMatrix<_Scalar> res(Matrix::_Rows);
|
||
|
|
_Scalar d = determinant();
|
||
|
|
for(int i = 0; i < Matrix::_Rows; i++) {
|
||
|
|
for (int j = 0; j < Matrix::_Cols; j++) {
|
||
|
|
res.Data(j, i) = 1.0*M(i, j)/d;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return res;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
};//class Matrix end
|
||
|
|
|
||
|
|
template<typename _Scalar>
|
||
|
|
_Scalar SMatrix<_Scalar>::determinant() {
|
||
|
|
int r, c, m;
|
||
|
|
int lop = 0;
|
||
|
|
int n = Matrix::_Rows;
|
||
|
|
_Scalar result = 0;
|
||
|
|
_Scalar mid = 1;
|
||
|
|
if (n != 1) {
|
||
|
|
lop = (n == 2) ? 1 : n;
|
||
|
|
for (m = 0; m < lop; m++) {
|
||
|
|
mid = 1;
|
||
|
|
for (r = 0, c = m; r < n; r++, c++) {
|
||
|
|
mid = mid * (*(Matrix::data+r*n+c%n));
|
||
|
|
}
|
||
|
|
result += mid;
|
||
|
|
}
|
||
|
|
for (m = 0; m < lop; m++) {
|
||
|
|
mid = 1;
|
||
|
|
for (r = 0, c = n-1-m+n; r < n; r++, c--) {
|
||
|
|
mid = mid * (*(Matrix::data+r*n+c%n));
|
||
|
|
}
|
||
|
|
result -= mid;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
result = Matrix::data[0];
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
template<typename _Scalar>
|
||
|
|
_Scalar SMatrix<_Scalar>::M(int m, int n) {
|
||
|
|
float mid_result = 0;
|
||
|
|
int sign = 1;
|
||
|
|
int k = Matrix::_Rows;
|
||
|
|
SMatrix mid(k-1);
|
||
|
|
int c = 0;
|
||
|
|
for (int i = 0; i < k; i++) {
|
||
|
|
for (int j = 0; j < k; j++) {
|
||
|
|
if (i != m && j != n)
|
||
|
|
{
|
||
|
|
mid.Data(c++) = Matrix::cData(i,j);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
sign = (m+n)%2 == 0 ? 1 : -1;
|
||
|
|
mid_result = (float)sign*mid.determinant();
|
||
|
|
return mid_result;
|
||
|
|
}
|
||
|
|
#undef Matrix
|
||
|
|
|
||
|
|
}//namespace Ctain end
|
||
|
|
#endif //MATRIX_SQUAREMATRIX_H
|