http://202.120.80.191/problem.php?problemid=1040
裸题
Description
应用高斯消元法求解n*n的线性方程组Ax=b,其中A为系数矩阵。
数据保证有唯一解。
Input
第1行为一个整数n(0<n<=20),表示是n*n的的线性方程组。
接下去的n行表示了系数矩阵A,每行有n个整数。
再接下去的n行表示了b,每行只有一个整数。
Output
输出有n行,每行有1个小数(精确到0.01),表示方程组的解。
Sample Input
3
1 2 3
2 4 5
3 1 2
4
3
1
Sample Output
-1.40
-4.80
5.00
ps:手写了个高斯消元 AC,超爽 ,注意要对double 0的情况特殊处理
#include<stdio.h> #include<math.h> #include<iostream> using namespace std; #define eps 1e-8 #define zero(a) fabs(a)<eps double map[29][29]; double ret[29]; void mat(int n){ int i,j,k,ti; for(k=1;k<n;k++){ int firsti=-1; double map_i_k; for(i=k;i<=n;i++){ if(firsti==-1){ if(zero(map[i][k])){ for(ti=i+1;ti<=n;ti++){ if(!zero(map[ti][k]))break; } if(ti==n+1)break; for(j=1;j<=(n+1);j++){ swap(map[i][j],map[ti][j]); } } firsti=i; } if(zero(map[i][k]))continue; map_i_k=map[i][k]; for(j=1;j<=(n+1);j++){ map[i][j]/=map_i_k; } if(firsti==-1)continue; if(firsti==i)continue; for(j=1;j<=(n+1);j++){ map[i][j]-=map[firsti][j]; } } } //test //for(i=1;i<=n;i++)ret[i]=0; //for(i=1;i<=n;i++){ // for(j=1;j<=n;j++){ // if(!zero(map[i][j]))break; // } // if(j==n+1)continue; // ret[j]=map[i][n+1]/map[i][j]; //} //for(j=1;j<=n;j++){ // printf("%.2lf ",ret[j]); //} //for(i=1;i<=n;i++){ // for(j=1;j<=(n+1);j++){ // printf("%.8lf ",map[i][j]); // }printf(" "); //} // for(i=n;i>=1;i--){ for(j=n;j>i;j--){ map[i][n+1]-=map[i][j]*ret[j]; } ret[i]=map[i][n+1]*1.0/map[i][i]; } } int main(){ int n; while(scanf("%d",&n)!=EOF){ int i,j; for(i=1;i<=n;i++){ for(j=1;j<=n;j++){ scanf("%lf",&map[i][j]); } } for(i=1;i<=n;i++){ scanf("%lf",&map[i][n+1]); } mat(n); for(i=1;i<=n;i++){ if(zero(ret[i]))printf("0.00 "); else printf("%.2lf ",ret[i]); } } return 0; }