【BZOJ1013】球形空间产生器sphere

高斯消元模板题

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>

using namespace std;

const int maxn=30;
const double EPS=1e-9;
double G[maxn][maxn];
double a[maxn][maxn];
double ans[maxn];
int n;
double g;
void gauss_jordan(){
    int num;
    for(int i=1;i<n;i++){
        num=i;
        for(int j=i+1;j<=n;j++){
            if(fabs(a[j][i])>fabs(a[num][i]))
                num=j;
        }
        for(int j=1;j<=n+1;j++)
            swap(a[i][j],a[num][j]);
        for(int j=i+1;j<=n;j++){
            if(fabs(a[j][i])<=EPS)continue;
            double t=a[j][i]/a[i][i];
            for(int k=i;k<=n+1;k++){
                a[j][k]-=a[i][k]*t;
            }
        }
    }
    for(int i=n;i>=1;i--){
        ans[i]=a[i][n+1]/a[i][i];
        for(int j=i-1;j>=1;j--){
            a[j][n+1]-=a[j][i]*ans[i];
        }
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n+1;i++){
        for(int j=1;j<=n;j++){
            scanf("%lf",&g);
            G[i][j]+=-2.0*g;
            G[i][n+1]+=g*g;
        }
    }
    for(int i=1;i<=n;i++){
            for(int l=1;l<=n+1;l++){
                a[i][l]=G[i][l]-G[i+1][l];
            }
            a[i][n+1]=-a[i][n+1];

    }

    /*for(int i=1;i<=n;i++){
        for(int j=1;j<=n+1;j++){
            printf("%.1f ",a[i][j]);
        }
        printf("
");
    }*/
    gauss_jordan();
    for(int i=1;i<=n;i++){
        printf("%.3f ",ans[i]);
    }

    return 0;
}