高斯消元 - JSOI 2008 球形空间产生器 - 洛谷 P4035

高斯消元 - JSOI 2008 球形空间产生器 - 洛谷 P4035

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

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

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

输入格式

第一行是一个整数n。

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

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

输出格式

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

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

数据范围

$1\le n\le 10$

输入样例：

2
0.0 0.0
-1.0 1.0
1.0 0.0

输出样例：

0.500 1.500

---

分析：

$设n维坐标下的球心为(x_1,x_2,...,x_n)，n+1个点的坐标为(a_{i1},a_{i2},...,a_{in}{)}_{1\le i\le n+1}，球半径为R，则：$

$\{\begin{array}{l}(a_{11}-x_1{)}^2+(a_{12}-x_2{)}^2+...+(a_{1n}-x_n{)}^2=R^2\\ \\ (a_{21}-x_1{)}^2+(a_{22}-x_2{)}^2+...+(a_{2n}-x_n{)}^2=R^2\\ \\ ...\\ \\ (a_{n+1,1}-x_1{)}^2+(a_{n+1,2}-x_2{)}^2+...+(a_{n+1,n}-x_n{)}^2=R^2\end{array}$

$本题看似是二次方程组，但是我们发现，每个方程的二次项x_i^2都对应相等，$

$故我们对任意两个方程作差，就能够得到关于x_i的线性方程。$

$本题，将所有方程减去第一个方程，得到新的线性方程组：$

$\{\begin{array}{l}{a}_{21}^2-a_{11}^2-2(a_{21}-a_{11})x_1+a_{22}^2-a_{12}^2-2(a_{22}-a_{12})x_2+...+a_{2n}^2-a_{1n}^2-2(a_{2n}-a_{1n})x_n=0\\ \\ a_{31}^2-a_{11}^2-2(a_{31}-a_{11})x_1+a_{32}^2-a_{12}^2-2(a_{32}-a_{12})x_2+...+a_{3n}^2-a_{1n}^2-2(a_{3n}-a_{1n})x_n=0\\ \\ ...\\ \\ a_{n+1,1}^2-a_{11}^2-2(a_{n+1,1}-a_{11})x_1+a_{n+1,2}^2-a_{12}^2-2(a_{n+1,2}-a_{12})x_2+...+a_{n+1,n}^2-a_{1n}^2-2(a_{n+1,n}-a_{1n})x_n=0\end{array}$

$等价于：$

$\{\begin{array}{l}2(a_{21}-a_{11})x_1+2(a_{22}-a_{12})x_2+...+2(a_{2n}-a_{1n})x_n=a_{21}^2-a_{11}^2+a_{22}^2-a_{12}^2+...+a_{2n}^2-a_{1n}^2\\ \\ 2(a_{31}-a_{11})x_1+2(a_{32}-a_{12})x_2+...+2(a_{3n}-a_{1n})x_n=a_{31}^2-a_{11}^2+a_{32}^2-a_{12}^2+...+a_{3n}^2-a_{1n}^2\\ \\ ...\\ \\ 2(a_{n+1,1}-a_{11})x_1+2(a_{n+1,2}-a_{12})x_2+...+2(a_{n+1,n}-a_{1n})x_n=a_{n+1,1}^2-a_{11}^2+a_{n+1,2}^2-a_{12}^2+...+a_{n+1,n}^2-a_{1n}^2\end{array}$

增广矩阵：
$$\left[\begin{array}{ccccc}2(a_{21}-a_{11})& 2(a_{22}-a_{12})& ...& 2(a_{2n}-a_{1n})& a_{21}^2-a_{11}^2+a_{22}^2-a_{12}^2+...+a_{2n}^2-a_{1n}^2\\ & & & & \\ 2(a_{31}-a_{11})& 2(a_{32}-a_{12})& ...& 2(a_{3n}-a_{1n})& a_{31}^2-a_{11}^2+a_{32}^2-a_{12}^2+...+a_{3n}^2-a_{1n}^2\\ & & & & \\ ...& ...& ...& ...& ...\\ & & & & \\ 2(a_{n+1,1}-a_{11})& 2(a_{n+1,2}-a_{12})& ...& 2(a_{n+1,n}-a_{1n})& a_{n+1,1}^2-a_{11}^2+a_{n+1,2}^2-a_{12}^2+...+a_{n+1,n}^2-a_{1n}^2\end{array}\right]$$

$令b_{ij}=a_{i+1,j}-a_{1j}，则b_{i,n+1}={\sum }_{j=1}^n{a}_{i+1,j}^2-a_{1j}^2，用高斯消元解方程组即可。$

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#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;c<=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;
        
        //交换
        for(int i=c;i<=n+1;i++) swap(b[r][i],b[t][i]);
        //归一化
        for(int i=n+1;i>=c;i--) b[r][i]/=b[r][c];
        //消成上三角阵
        for(int i=r+1;i<=n;i++) 
            for(int j=n+1;j>=c;j--)
                b[i][j]-=b[i][c]*b[r][j];
    }
    
    //转化成对角阵
    for(int i=n;i>1;i--)    //i为列，j为行
        for(int j=i-1;j;j--)
        {
            b[j][n+1]-=b[i][n+1]*b[j][i];
            b[j][i]=0;
        }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n+1;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+1][j]-a[1][j]);
            b[i][n+1]+=a[i+1][j]*a[i+1][j]-a[1][j]*a[1][j];
        }
        
    gauss();
    
    for(int i=1;i<=n;i++) printf("%.3lf ",b[i][n+1]);
    
    return 0;
}