高斯消元[模版]

题面描述

有$n$个未知数$x_1$,$x_2$,$x_3$……$x_n$，满足：

$$a_{1,1}\ast x_1+a_{1,2}\ast x_2+a_{1,3}\ast x_3\dots \dots +a_{1,n}\ast x_n=b_1$$
$$a_{2,1}\ast x_1+a_{2,2}\ast x_2+a_{2,3}\ast x_3\dots \dots +a_{2,n}\ast x_n=b_2$$
$$\dots \dots \dots \dots \dots \dots \dots \dots \dots \dots \dots \dots \dots \dots$$
$$a_{n,1}\ast x_1+a_{n,2}\ast x_2+a_{n,3}\ast x_3\dots \dots +a_{n,n}\ast x_n=b_n$$
求$x_1,x_2,x_3\dots \dots x_n$的值。（保证有解）
【输入格式】
第一行给出$n(1<=n<=100)$
下来$n$行，每行给出$n+1$个实数，分别是 $a_{i,1}$~$a_{i,n}$ 和$b_i$。
【输出格式】
输出$x_1,x_2,x_3\dots \dots x_n$的值（相邻两个用一个空格隔开，每个数保留$3$位小数）
【样例输入】
3
2.5 5.0 3.0 32.5
1.0 4.5 2.0 22
4.0 3.5 1.5 26.5
【样例输出】
3.000 2.000 5.000

思路

作为高斯消元法的模版题，主要有两种初中的消元方法，分别是加减消元，以及代入消元。
实际步骤：
如图,
$\left|\begin{array}{cccc}2.5& 5.0& 3.0|& 32.5\\ 1.0& 4.5& 2.0|& 22\\ 4.0& 3.5& 1.5|& 26.5\end{array}\right|$
$\Rrightarrow$
$\left|\begin{array}{cccc}2.5& 5.0& 3.0|& 32.5\\ 0& 2.5& 0.8|& 9\\ 0& -4.5& -3.3|& -25.5\end{array}\right|$
$\Rrightarrow$
$\left|\begin{array}{cccc}2.5& 5.0& 3.0|& 32.5\\ 0& 2.5& 0.8|& 9\\ 0& 0& -1.86|& -9.3\end{array}\right|$
$\Rrightarrow$
$x_3=-9.3/(-1.86)=5.000$
$x_2=[9-(0.8\ast x_3)]/2.5=2.000$
$x_1=[32.5-15-10]/2.5=3.000$

1. 先从$1$~$n$枚举每一行，对于$\forall x,i<x\le n$，削去$x$行$i$列的系数，使系数成为$0$
2. 观察可以发现，最后一行$n$只有$x_n$的系数不为0，则可求出$x_n$，进而代入消元。

具体过程见代码。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
using namespace std;
const int N=110;
double a[N][N];
double f[N];int n;
void gauss()
{
   for(int i=1;i<=n;i++)
   {
   	int r=i;//仅仅只是为了减少精度损失
   	for(int j=i+1;j<=n;j++)if(fabs(a[j][i])>fabs(a[r][i]))r=j;
   	for(int j=1;j<=n+1;j++){double t=a[r][j];a[r][j]=a[i][j];a[i][j]=t;}
   	for(int j=n+1;j>=i;j--)
   	{
   		for(int k=i+1;k<=n;k++)
   			a[k][j]-=a[k][i]/a[i][i]*a[i][j];
   	}
   }
   for(int i=n;i>=1;i--)
   {
   	double t=a[i][n+1];
   	for(int j=i+1;j<=n;j++)
   	{
   		t-=a[i][j]*f[j];
   	}
   	f[i]=t/a[i][i];
   }
}
int main()
{
   scanf("%d",&n);
   for(int i=1;i<=n;i++)for(int j=1;j<=n+1;j++)scanf("%lf",&a[i][j]);
   gauss();
   for(int i=1;i<=n;i++)printf("%.3lf ",f[i]);puts("");
   return 0;
}