【算法基础17】如何用c++做线性代数作业？高斯消元的妙用

高斯消元

        主要思想：将方程组系数矩阵转化为梯度矩阵，在O()内解n元高次方程组。每次找到未处理过的行里绝对值最大的一行作为基准，将这一行里的每个系数都除以a[r][c]，使得a[r][c]的值为1，再将其他行都减去a[i][c]倍的基准行，使得其他行的a[i][c]都化为0，最终得到了一个每一行第一个不为0的系数都为1的梯度矩阵。由第n行能解出Xn的值，代入第n-1行能解出Xn-1的值，最终得出方程组的所有解。

        例题：给出一个n元高次方程组，求方程组的解。

#include<iostream>
#include<cmath>
using namespace std;

const int N=1010;
const double eps=1e-6;//浮点数用一个极小的数判断是不是为0

int n;
double a[N][N];

int gauss(){
	int c,r;
	for(c=0,r=0;c<n;c++){
		int t=r;
		for(int i=r;i<n;i++){
			if(fabs(a[t][c])<fabs(a[i][c]))//找出绝对值最大的一行
				t=i;
		}		
		if(fabs(a[t][c])<eps) continue;//绝对值为0，不用转化，处理下一行
		
		for(int i=c;i<n+1;i++) swap(a[t][i],a[r][i]);//交换到第r行，注意循环条件
		for(int i=n;i>=c;i--) a[r][i]/=a[r][c];//将a[r][c]化为1
		
		for(int i=r+1;i<n;i++){//以第r行为基准，处理其他行
			if(fabs(a[i][c])>eps){
				for(int j=n;j>=c;j--){//j从n开始，即第n+1列，即第i个方程的解
					a[i][j]-=a[i][c]*a[r][j];
				}
			}
		}
		r++;
	
	}
	
	if(r<n){//秩小于n，方程组无解或有无穷解
		for(int i=r;i<n;i++){
			if(fabs(a[i][n])>eps) return 2;//无解 
		}
		return 1;//无穷解 
	}
	
	for(int i=n-1;i>=0;i--){//倒序遍历，代入解出所有解
		for(int j=i+1;j<n;j++)
			a[i][n]-=a[i][j]*a[j][n];
	}
	return 0;//唯一解 
}

int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		for(int j=0;j<n+1;j++)
			cin>>a[i][j];
	}
	
	int t=gauss();
	
	if(t==0){
		for(int i=0;i<n;i++)
			printf("%.2lf\n",a[i][n]);
	}
	else if(t==2) cout<<'no solution';
	else cout<<'infinite group solutions';

	return 0;
}

         注：n元高次方程组的系数矩阵为大小为n*(n+1)，按列遍历的时候注意判定条件。