📄 matrix.inl
字号:
break; //return mRank;
else
mRank++; //子阵元素绝对值最大者不为0,矩阵秩加1
if(k ==(ColRowMin - 1)) //已到最后一行(列)
break; //return mRank;
if(iSign != k) //主元不在当前行
{
for(size_t j = k; j < stCol; j ++) //交换行元素
swap(m(k, j), m(iSign, j));
}
if(jSign != k) //主元不在当前列
{
for(size_t i = k; i < stRow; i ++) //交换列元素
swap(m(i, jSign), m(i, k));
}
for(i = k + 1; i < stRow; i ++)
{
const _Ty d(m(i, k) / m(k, k)); //消元因子
for(size_t j = k + 1; j < stCol; j ++)
m(i, j) -= d * m(k, j); //当前主元右下阵元素作变换
}
}
return mRank;
}
//全选主元高斯-约当(Gauss-Jordan)法求矩阵逆
//矩阵逆取代原来的输入矩阵
template <class _Ty>
int MatrixInversionGS(matrix<_Ty >& rhs)
{
size_t stRank = rhs.GetColNum(); // 矩阵阶数
if(stRank != rhs.GetRowNum())
return int(-1); //rhs不是方阵
valarray<size_t> is(stRank); //行交换信息
valarray<size_t> js(stRank); //列交换信息
matrix<_Ty> m(rhs); //生成一matrix对象
for(size_t k = 0; k < stRank; k++)
{ // 全选主元
long double MaxValue(0);
for(size_t i = k; i < stRank; i ++)
{
for(size_t j = k; j < stRank; j ++)
{
long double tmp(Abs(m(i, j))); //求m(i,j)绝对值
if(tmp > MaxValue) //主元不在对角线上
{
MaxValue = tmp;
is[k] = i; //记录主元行数
js[k] = j; //记录主元列数
}
}
}
if(FloatEqual(MaxValue, 0))
return int(0); //主元为0,矩阵奇异
if(is[k] != k) //主元不在当前行
{
for(size_t j = 0; j < stRank; j ++) //交换行元素
swap(m(k, j), m(is[k], j));
}
if(js[k] != k) //主元不在当前列
{
for(size_t i = 0; i < stRank; i ++) //交换列元素
swap(m(i, k), m(i, js[k]));
}
m(k, k) = 1.0 / m(k, k); //主元倒数
for(size_t j = 0; j < stRank; j ++)
if(j != k)
m(k, j) *= m(k, k);
for(i = 0; i < stRank; i ++)
if(i != k)
for(size_t j = 0; j < stRank; j ++)
if(j != k)
m(i, j) = m(i, j) - m(i, k) * m(k, j);
for(i = 0; i < stRank; i ++)
if(i != k)
m(i, k) = -m(i, k) * m(k, k);
}
for(int r = stRank - 1; r >= 0; r--)
{
if(js[r] != r)
for(size_t j = 0; j < stRank; j ++)
swap(m(r, j), m(js[r], j));
if(is[r] != r)
for(size_t i = 0; i < stRank; i ++)
swap(m(i, r), m(i, is[r]));
}
rhs = m;
return int(1);
}
//用“变量循环重新编号法”求对称正定矩阵逆
//矩阵类型必须是浮点型
template <class _Ty>
int MatrixSymmetryRegularInversion(matrix<_Ty>& rhs)
{
int iSym = MatrixSymmetryRegular(rhs, 1); //判别是否对称正定
if(iSym < 2)
return (iSym); //rhs不是对称正定阵
size_t stRank = rhs.GetColNum(); // 矩阵阶数
matrix<_Ty> msr(rhs); //生成一matrix对象,用rhs初始化
valarray<_Ty> b(stRank);
for(size_t k=0; k<stRank; k++)
{
_Ty w= msr(0, 0);
size_t m = stRank - k -1;
for(size_t i = 1; i < stRank; i++)
{
_Ty g = msr(i, 0);
b[i] = g / w;
if(i <= m) b[i] = -b[i];
for(size_t j = 1; j <= i; j ++)
msr((i-1),(j-1)) = msr(i, j) + g * b[j];
}
msr(stRank-1, stRank-1) = 1.0 / w;
for(i = 1; i < stRank; i ++)
msr(stRank-1,(i-1)) = b[i];
}
for(size_t i = 0; i < stRank-1; i ++)
for(size_t j = i+1; j < stRank; j ++)
msr(i,j) = msr(j, i);
rhs = msr;
return (iSym);
}
//特兰持(Trench)法求托伯利兹(Toeplitz)矩阵逆
//矩阵类型必须是浮点型
template <class _Ty>
int MatrixToeplitzInversionTrench(const valarray<_Ty>& t, const valarray<_Ty>& tuo, matrix<_Ty>& rhs)
{
size_t stRank = rhs.GetColNum(); // 矩阵阶数
if(stRank != rhs.GetRowNum())
return int(-1); //rhs不是方阵
if(FloatEqual(t[0], 0))
return int(0);
valarray<_Ty> c(stRank);
valarray<_Ty> r(stRank);
valarray<_Ty> p(stRank);
_Ty a=t[0];
c[0]=tuo[1]/t[0];
r[0]=t[1]/t[0];
matrix<_Ty> b(rhs);
for(size_t k=0; k<=stRank-3; k++)
{
_Ty s=0.0;
for(size_t j=1; j<=k+1; j++)
s=s+c[k+1-j]*tuo[j];
s=(s-tuo[k+2])/a;
for(size_t i=0; i<=k; i++)
p[i]=c[i]+s*r[k-i];
c[k+1]=-s;
s=0.0;
for(j=1; j<=k+1; j++)
s=s+r[k+1-j]*t[j];
s=(s-t[k+2])/a;
for(i=0; i<=k; i++)
{
r[i]=r[i]+s*c[k-i];
c[k-i]=p[k-i];
}
r[k+1]=-s;
a=0.0;
for(j=1; j<=k+2; j++)
a=a+t[j]*c[j-1];
a=t[0]-a;
if(FloatEqual(a, 0))
return int(0);
}
b(0,0)=1.0/a;
for(size_t i=0; i<stRank-1; i++)
{
k=i+1;
b(0, k)=-r[i]/a;
b(i+1,0)=-c[i]/a;
}
for(i=0; i<stRank-1; i++)
for(size_t j=0; j<stRank-1; j++)
b(i+1, j+1)=b(i,j)-c[i]*b(0,j+1)+c[stRank-j-2]*b(0,stRank-i-1);
rhs = b;
return int(1);
}
//实矩阵LU分解
//矩阵必须是浮点型
//其中,第一参数是输入矩阵,必须是方阵,不一定对称,第二个是单元下三角矩阵L,第三个是上三角矩阵
template <class _Ty>
int MatrixLU(const matrix<_Ty>& rhs, matrix<_Ty>& lhs, matrix<_Ty>& uhs)
{
size_t stRank = rhs.GetColNum(); // 矩阵阶数
if(stRank != rhs.GetRowNum())
return int(-1); //rhs不是方阵
matrix<_Ty> m(rhs);
for(size_t k=0; k<stRank-1; k++)
{
if(FloatEqual(m(k,k),0))
return int(0); //主元为0
for(size_t i=k+1; i<stRank; i++)
m(i,k) /= m(k,k);
for(i=k+1; i<stRank; i++)
for(size_t j=k+1; j<stRank; j++)
m(i,j)=m(i,j)-m(i,k)*m(k,j);
}
//给上、下三角阵赋值
for(size_t i=0; i<stRank; i++)
{
for(size_t j=0; j<i; j++)
{
lhs(i,j)=m(i,j);
uhs(i,j)=0.0;
}
lhs(i,i)=1.0;
uhs(i,i)=m(i,i);
for(j=i+1; j<stRank; j++)
{
lhs(i,j)=0.0;
uhs(i,j)=m(i,j);
}
}
return (1); //分解成功
}
//用豪斯荷尔德(Householder)变换对一般m*n阶的实矩阵进行QR分解
//输入矩阵matA, 分解后matA变成R,matQ是函数第二个参数
template <class _Ty>
int MatrixQR(matrix<_Ty>& rhs, matrix<_Ty>& rhq)
{
size_t stRow = rhs.GetRowNum(); // 矩阵行数
size_t stCol = rhs.GetColNum(); // 矩阵列数
if(stRow < stCol)
return (0); //行不能小于列
for(size_t i=0; i<stRow; i++)
for(size_t j=0; j<stRow; j++)
{
rhq(i,j)=0.0;
if(i==j) rhq(i,j)=1.0;
}
size_t nn=stCol;
if(stRow == stCol) nn=stRow-1;
for(size_t k=0; k<nn; k++)
{
_Ty u=0.0;
for(size_t i = k; i < stRow; i++)
{
_Ty w = Abs(rhs(i,k));
if(w > u) u = w;
}
_Ty alpha=0.0;
for(i = k; i < stRow; i++)
{
_Ty t=rhs(i,k)/u;
alpha=alpha+t*t;
}
if(rhs(k,k)>0.0) u=-u;
alpha=u*sqrt(alpha);
if(FloatEqual(alpha,0.0)) return(0);
u=sqrt(2.0*alpha*(alpha-rhs(k,k)));
if(FloatNotEqual(u,0.0))
{
rhs(k,k)=(rhs(k,k)-alpha)/u;
for(i=k+1; i<stRow; i++)
rhs(i,k) /= u;
for(size_t j=0; j<stRow; j++)
{
_Ty t=0.0;
for(size_t jj=k; jj<stRow; jj++)
t=t+rhs(jj,k)*rhq(jj,j);
for(i=k; i<stRow; i++)
rhq(i,j)=rhq(i,j)-2.0*t*rhs(i,k);
}
for(j=k+1; j<stCol; j++)
{
_Ty t=0.0;
for(size_t jj=k; jj<stRow; jj++)
t=t+rhs(jj,k)*rhs(jj,j);
for(i=k; i<stRow; i++)
rhs(i,j)=rhs(i,j)-2.0*t*rhs(i,k);
}
rhs(k,k)=alpha;
for(i=k+1; i<stRow; i++)
rhs(i,k)=0.0;
}
}
for(i=0; i<stRow-1; i++)
for(size_t j=i+1; j<stRow;j++)
swap(rhq(i,j), rhq(j,i));
return (1);
}
//对称正定阵的乔里斯基(Cholesky)分解及求其行列式值
//输入矩阵必须是对称正定矩阵;计算量大为减少,大约是克劳特分解的工作量一半
//矩阵与返回值类型必须是浮点型
template <class _Ty>
long double MatrixSymmetryRegularCholesky(matrix<_Ty>& rhs)
{
int iReValue= MatrixSymmetryRegular(rhs, 1); //判别对称正定
if(iReValue < 2)
return long double(0); //rhs不是对称正定阵
size_t stRank = rhs.GetColNum(); // 矩阵阶数
matrix<_Ty> m(rhs); //生成一matrix对象,用rhs初始化
long double Det = m(0,0); //计算行列式值
m(0,0) = sqrt(m(0,0));
for(size_t i=1; i<stRank; i++)
m(i, 0) /= m(0, 0);
for(size_t j=1; j<stRank; j++)
{
for(size_t k=0; k<j; k++)
m(j,j) = m(j,j) - m(j,k) * m(j,k);
Det *= m(j,j);
m(j,j) = sqrt(m(j,j));
for(i=j+1; i<stRank; i++)
{
for(k=0; k<j; k++)
m(i,j) = m(i,j) -m(i,k) * m(j,k);
m(i,j) /= m(j,j);
}
}
for(i=0; i<stRank-1; i++)
for(j=i+1; j<stRank; j++)
m(i,j)=0;
rhs = m; //返回Cholesky阵,原矩阵将被复盖
return Det; //返回行列式值
}
//一般实矩阵的奇异值分解
//将矩阵A分解成U*S*V
//其中,矩阵A作为第一个参数输入,分解完后矩阵A将变成第二个参数S,第一个参数表示U,第三个参数表示V
//值得说明的是,matlab里是将A分解成U*S*V';那么这里就不用再转置了
template <class _Ty>
int MatrixSingularValue(matrix<_Ty>& a, matrix<_Ty>& u,
matrix<_Ty>& v, _Ty eps)
{
int i, it(60), kk, mm, nn, m1, ks, ka;
_Ty d,dd,t,sm,sm1,em1,sk,ek,b,c,shh;
int m = a.GetRowNum();
int n = a.GetColNum();
for(int j=0; j<m; j++) u(j, m-1) = _Ty(0);
if(m > n) ka = m + 1;
else ka = n + 1;
valarray<_Ty> s(ka), e(ka), w(ka), fg(2), cs(2);
int k = n;
if(m-1<n) k=m-1;
int l = m;
if(n-2<m) l=n-2;
if(l<0) l=0;
int ll=k;
if(l>k) ll=l;
if(ll>=1)
{
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -