矩阵求逆的快速算法

算法介绍

矩阵求逆在3D程序中很常见,主要应用于求Billboard矩阵。按照定义的计算方法乘法运算,严重影响了性能。在需要大量Billboard矩阵运算时,矩阵求逆的优化能极大提高性能。这里要介绍的矩阵求逆算法称为全选主元高斯-约旦法。

高斯-约旦法(全选主元)求逆的步骤如下:

首先,对于 k 从 0 到 n - 1 作如下几步:

  1. 从第 k 行、第 k 列开始的右下角子阵中选取绝对值最大的元素,并记住次元素所在的行号和列号,在通过行交换和列交换将它交换到主元素位置上。这一步称为全选主元。
  2. m(k, k) = 1 / m(k, k)
  3. m(k, j) = m(k, j) * m(k, k),j = 0, 1, ..., n-1;j != k
  4. m(i, j) = m(i, j) - m(i, k) * m(k, j),i, j = 0, 1, ..., n-1;i, j != k
  5. m(i, k) = -m(i, k) * m(k, k),i = 0, 1, ..., n-1;i != k

最后,根据在全选主元过程中所记录的行、列交换的信息进行恢复,恢复的原则如下:在全选主元过程中,先交换的行(列)后进行恢复;原来的行(列)交换用列(行)交换来恢复。

实现(4阶矩阵)

float Inverse(CLAYMATRIX& mOut, const CLAYMATRIX& rhs)
{
	CLAYMATRIX m(rhs);
	DWORD is[4];
	DWORD js[4];
	float fDet = 1.0f;
	int f = 1;

	for (int k = 0; k < 4; k ++)
	{
		// 第一步,全选主元
		float fMax = 0.0f;
		for (DWORD i = k; i < 4; i ++)
		{
			for (DWORD j = k; j < 4; j ++)
			{
				const float f = Abs(m(i, j));
				if (f > fMax)
				{
					fMax	= f;
					is[k]	= i;
					js[k]	= j;
				}
			}
		}
		if (Abs(fMax) < 0.0001f)
			return 0;
		
		if (is[k] != k)
		{
			f = -f;
			swap(m(k, 0), m(is[k], 0));
			swap(m(k, 1), m(is[k], 1));
			swap(m(k, 2), m(is[k], 2));
			swap(m(k, 3), m(is[k], 3));
		}
		if (js[k] != k)
		{
			f = -f;
			swap(m(0, k), m(0, js[k]));
			swap(m(1, k), m(1, js[k]));
			swap(m(2, k), m(2, js[k]));
			swap(m(3, k), m(3, js[k]));
		}

		// 计算行列值
		fDet *= m(k, k);

		// 计算逆矩阵

		// 第二步
		m(k, k) = 1.0f / m(k, k);	
		// 第三步
		for (DWORD j = 0; j < 4; j ++)
		{
			if (j != k)
				m(k, j) *= m(k, k);
		}
		// 第四步
		for (DWORD i = 0; i < 4; i ++)
		{
			if (i != k)
			{
				for	(j = 0; j < 4; j ++)
				{
					if (j != k)
						m(i, j) = m(i, j) - m(i, k) * m(k, j);
				}
			}
		}
		// 第五步
		for (i = 0; i < 4; i ++)
		{
			if (i != k)
				m(i, k) *= -m(k, k);
		}
	}

	for	(k = 3; k >= 0; k --)
	{
		if (js[k] != k)
		{
			swap(m(k, 0), m(js[k], 0));
			swap(m(k, 1), m(js[k], 1));
			swap(m(k, 2), m(js[k], 2));
			swap(m(k, 3), m(js[k], 3));
		}
		if (is[k] != k)
		{
			swap(m(0, k), m(0, is[k]));
			swap(m(1, k), m(1, is[k]));
			swap(m(2, k), m(2, is[k]));
			swap(m(3, k), m(3, is[k]));
		}
	}

	mOut = m;
	return fDet * f;
}

比较

  原算法 原算法(经过高度优化) 新算法
加法次数 103 61 39
乘法次数 170 116 69
需要额外空间 16 * sizeof(float) 34 * sizeof(float) 25 * sizeof(float)

结果不言而喻吧。

你可能感兴趣的:(矩阵求逆的快速算法)