高斯消元模板,求对一个n为矩阵进行消元求出n个解

#include<bits/stdc++.h>
using namespace std;
double a[20][20],b[20],c[20][20];
int n;
int main(){
    cin>>n;
    for(int i=1;i<=n+1;i++){
        for(int j=1;j<=n;j++){
            scanf("%lf",&a[i][j]);
        }
    }
    //先表示出这个矩阵中的各项以及它们的系数  
    //这里c表示矩阵里面的各项,b表示最后一列的常数项 
    for(int i=1;i<=n;i++){ 
        for(int j=1;j<=n;j++){
            c[i][j]=2*(a[i][j]-a[i+1][j]);
            b[i]+=a[i][j]*a[i][j]-a[i+1][j]*a[i+1][j];
        }
    }
    //进行高斯消元,其目的就是将矩阵化为一个对角线为1的矩阵 
    for(int i=1;i<=n;i++){
        //找到x的系数不为0的方程 
        for(int j=i;j<=n;j++){
            if(fabs(c[j][i])>1e-8){
                for(int k=1;k<=n;k++)  swap(c[i][k],c[j][k]);
                swap(b[i],b[j]);
            }
        }
        //找到后消去其他方程的x的系数 
        for(int j=1;j<=n;j++){
            if(i==j)  continue;
            double rate=c[j][i]/c[i][i];
            for(int k=i;k<=n;k++)  c[j][k]-=c[i][k]*rate;
            b[j]-=b[i]*rate; 
        }
    }
    for(int i=1;i<n;i++)  printf("%.3lf ",b[i]/c[i][i]);
    printf("%.3lf\n",b[n]/c[n][n]);
    return 0;
}
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务