題目戳這裏
高斯消元模板
輸入格式如下:
n
a1 a2 a3 … an b1
共n個線性方程構成的方程組
輸出:ans[i]表示x[i]
/*
Gauss消元
給定線性方程組
a11*x1+a12*x2+...+a1n*xn=b1
a21*x1+a22*x2+...+a2n*xn=b2
...
an1*x1+an2*x2+...+ann*xn=bn
求x1,x2...xn
O(n^3)
*/
#include <bits/stdc++.h>
using namespace std;
double mp[111][111];
double ans[111];
double eps=1e-7;
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n+1;j++)
{
scanf("%lf",&mp[i][j]);
}
}
for(int i=1;i<=n;i++)
{
int r=i;
for(int j=i+1;j<=n;j++)
{
if(fabs(mp[r][i])<fabs(mp[j][i])) r=j;//find_the_biggest_number_of_the_first_column(at present)
}
if(fabs(mp[r][i])<eps)
{
printf("No Solution");
return 0;
}
if(i!=r) swap(mp[i],mp[r]);//對換一行或一列,屬於找最大當前係數的其中一步。(這樣就可以只處理當前行的係數啦!)
double div=mp[i][i];
for(int j=i;j<=n+1;j++) mp[i][j]/=div;
for(int j=i+1;j<=n;j++)
{
div=mp[j][i];
for(int k=i;k<=n+1;k++) mp[j][k]-=mp[i][k]*div;
}
}
ans[n]=mp[n][n+1];
for(int i=n-1;i>=1;i--)
{
ans[i]=mp[i][n+1];
for(int j=i+1;j<=n;j++) ans[i]-=(mp[i][j]*ans[j]);
}//迴帶操作
for(int i=1;i<=n;i++) printf("%.2lf\n",ans[i]);
}