高斯消元模板

来源:互联网 发布:淘宝宝贝描述低能买吗 编辑:程序博客网 时间:2024/06/05 21:08
#include<bits/stdc++.h>using namespace std;#define eps 1e-9const int MAXN=220;const int N=3;double a[MAXN][MAXN],x[MAXN];//方程的左边的矩阵和等式右边的值,求解之后x存的就是结果int equ,var;//方程数和未知数个数/**返回0表示无解,1表示有解*/int Gauss()//高斯消元 { int i,j,k,col,max_r; for(k=0,col=0;k<equ&&col<var;k++,col++) { max_r=k; for(i=k+1;i<equ;i++) if(fabs(a[i][col])>fabs(a[max_r][col])) max_r=i; if(fabs(a[max_r][col])<eps)return 0; if(k!=max_r) { for(j=col;j<var;j++) swap(a[k][j],a[max_r][j]); swap(x[k],x[max_r]); } x[k]/=a[k][col]; for(j=col+1;j<var;j++)a[k][j]/=a[k][col]; a[k][col]=1; for(i=0;i<equ;i++) if(i!=k) { x[i]-=x[k]*a[i][k]; for(j=col+1;j<var;j++)a[i][j]-=a[k][j]*a[i][col]; a[i][col]=0; } } return 1;}int main(){while(1){equ=var=N;for(int i=0;i<N;i++){for(int j=0;j<N;j++) {scanf("%lf",&a[i][j]);}scanf("%lf",&x[i]);}if(Gauss()) for(int i=0;i<N;i++) printf("x[%d]=%lf\n",i,x[i]);else printf("无解\n");}}