AcWing 207. 球形空间产生器（高斯消元）

题目描述

有一个球形空间产生器能够在 n 维空间中产生一个坚硬的球体。

现在，你被困在了这个 n 维球体中，你只知道球面上 n+1 个点的坐标，你需要以最快的速度确定这个 n 维球体的球心坐标，以便于摧毁这个球形空间产生器。

注意： 数据保证有唯一解。

输入格式

第一行是一个整数 n。

接下来的 n+1 行，每行有 n 个实数，表示球面上一点的 n 维坐标。

每一个实数精确到小数点后 6 位，且其绝对值都不超过 20000。

输出格式

有且只有一行，依次给出球心的 n 维坐标（n 个实数），两个实数之间用一个空格隔开。

每个实数精确到小数点后 3 位。

数据范围

1≤n≤10

输入样例：

2
0.0 0.0
-1.0 1.0
1.0 0.0

输出样例：

0.500 1.500

思路：

这是一道高斯消元的模板题，高斯消元就类似于线性代数化行阶梯的过程，就是先找这一行的对角线上这一列的最大值，然后再将找到的这一行跟现在遍历到的这一行交换，然后就是把这一行之下的所有行的这一列都变成0，不懂的小伙伴可以看一下线性代数，下面具体就看代码吧，把这个过程在纸上演算一下就行啦，下面请看代码：

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;

const int N = 15;

int n;
double a[N][N], b[N][N];

void Gauss() {
    //化成上三角矩阵
    for (int r = 1, c = 1; r <= n; ++ r, ++ c) {
        //找到主元
        int t = r;
        for (int i = r + 1; i <= n; ++ i)
            if (fabs(b[i][c]) > fabs(b[t][c]))
                t = i;
        //交换第r行和第t行的元素
        for (int i = c; i <= n + 1; ++ i) swap(b[r][i], b[t][i]);
        //主元归一（第r行除以主元系数）
        for (int i = n + 1; i >= c; -- i) b[r][i] /= b[r][c];
        //消元（用该行把下面所有行的第c列消为0）
        for (int i = r + 1; i <= n; ++ i) {
            for (int j = n + 1; j >= c; -- j) {
                b[i][j] -= b[r][j] * b[i][c];
            }
        }
    }
    //化成行最简阶梯型矩阵（本题唯一解，故同样也是对角矩阵）,把第j行的所有第i列的元素置为零 
    for (int i = n; i > 1; -- i) {
        for (int j = i - 1; j >= 1; -- j) {
            b[j][n + 1] -= b[i][n + 1] * b[j][i];
            b[j][i] = 0;
        }
    }
}
int main() {
    scanf("%d", &n);
    for (int i = 0; i <= n; ++ i) {
        for (int j = 1; j <= n; ++ j) {
            scanf("%lf", &a[i][j]);
        }
    }
    //转换为线性方程组
    for (int i = 1; i <= n; ++ i) {
        for (int j = 1; j <= n; ++ j) {
            b[i][j] = 2 * (a[i][j] - a[0][j]);
            b[i][n + 1] += a[i][j] * a[i][j] - a[0][j] * a[0][j];
        }
    }
    //Gauss消元
    Gauss();
    //输出答案
    for (int i = 1; i <= n; ++ i)
        printf("%.3lf ", b[i][n + 1]);
    return 0;
}