【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]);
}
发布了159 篇原创文章 · 获赞 13 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/KIKO_caoyue/article/details/99762486