BZOJ1013 JSOI2008 球形空间产生器sphere 高斯消元

#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define eps 1e-6

const int MAXN=10+2;
int N;
double p[MAXN],a[MAXN][MAXN],b[MAXN],x[MAXN];

void Gauss(){
    for(int i=1,p=i;i<=N;i++,p=i){
        for(int j=i+1;j<=N;j++)
            if(fabs(a[j][i])>fabs(a[p][i])) p=j;
        for(int j=1;j<=N;j++) swap(a[p][j],a[i][j]);
        swap(b[p],b[i]);

        for(int j=i+1;j<=N;j++){
            double r=a[j][i]/a[i][i];
            for(int k=1;k<=N;k++) a[j][k]-=a[i][k]*r;
            b[j]-=b[i]*r;
        }
    }

    for(int i=N;i;i--){
        x[i]=b[i]/a[i][i];
        for(int j=i-1;j;j--) b[j]-=x[i]*a[j][i],a[j][i]=0;
    }
}

int main(){
    cin >> N;
    for(int i=1;i<=N;i++) cin >> p[i];
    for(int i=1;i<=N;i++){
        double q;
        for(int j=1;j<=N;j++){
            cin >> q;
            a[i][j]=2*(p[j]-q),b[i]+=(p[j]*p[j]-q*q);
        }
    }

    Gauss();
    for(int i=1;i<=N;i++){
        printf("%.3lf",x[i]);
        if(i!=N) cout << " ";
    }

    return 0;
}