【Gauss消元】求線性方程組的解 洛谷P3389

題目戳這裏
高斯消元模板
輸入格式如下:
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]);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章