BZOJ 1013 [JSOI2008]球形空间产生器sphere 高斯消元

Description

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

Input

　　第一行是一个整数n(1<=N=10)。接下来的n+1行，每行有n个实数，表示球面上一点的n维坐标。每一个实数精确到小数点
后6位，且其绝对值都不超过20000。

Output

　　有且只有一行，依次给出球心的n维坐标（n个实数），两个实数之间用一个空格隔开。每个实数精确到小数点
后3位。数据保证有解。你的答案必须和标准输出一模一样才能够得分。

Sample Input

2
0.0 0.0
-1.0 1.0
1.0 0.0

Sample Output

0.500 1.500

HINT

　　提示：给出两个定义：1、 球心：到球面上任意一点距离都相等的点。2、 距离：设两个n为空间上的点A, B

的坐标为(a1, a2, …, an), (b1, b2, …, bn)，则AB的距离定义为：dist = sqrt( (a1-b1)^2 + (a2-b2)^2 +

… + (an-bn)^2 )

传送门

根据hint，假设球心坐标是x[1..n]，

以a[1],a[2]为例，那么我们可以得到：

(a[1,1]-x[1])^2+(a[1,2]-x[2])^2+……+(a[1,n]-x[n])^2=

(a[2,1]-x[1])^2+(a[2,2]-x[2])^2+……+(a[2,n]-x[n])^2

我们把式子展开，然后xi^2的项都可以约掉；

接着我们把所有和x[i]有关的项移到左边，其它移到右边，

得到：

2*(a[2,1]-a[1,1])*x[1]+2*(a[2,2]-a[1,2])*x[2]+……+2*(a[2,n]-a[1,n])*x[n]=

a[2,1]^2+a[2,2]^2+……+a[2,n]^2-a[1,1]^2-a[1,2]^2-……-a[1,n]^2

我们看到其实有n个方程，形式都是上面这样的，

第t个方程是：

2*(a[t+1,1]-a[t,1])*x[1]+2*(a[t+1,2]-a[t,2])*x[2]+……+2*(a[t+1,n]-a[t,n])*x[n]=

a[t+1,1]^2+a[t+1,2]^2+……+a[t+1,n]^2-a[t,1]^2-a[t,2]^2-……-a[t,n]^2

问题转化为求一次方程组的解。

当然首先先建立一下这个方程组。

接下来用高斯消元即可轻松搞定。

#include<bits/stdc++.h>
using namespace std;
const int 
	N=15;
int n;
double a[N][N],b[N],ans[N];
void getEquation(){
	for (int i=1;i<=n;i++){
		double sum=0.0;
		for (int j=1;j<=n;j++){
			sum+=a[i+1][j]*a[i+1][j]-a[i][j]*a[i][j];
			a[i][j]=2.0*(a[i+1][j]-a[i][j]);
		}
		b[i]=sum;
	}
}
void gauss(){
	double c[N];
	for (int k=1;k<n;k++){
		for (int i=k+1;i<=n;i++) c[i]=a[i][k]/a[k][k];
		for (int i=k+1;i<=n;i++){
			for (int j=1;j<=n;j++) a[i][j]-=a[k][j]*c[i];
			b[i]-=b[k]*c[i];
		}
	}
	ans[n]=b[n]/a[n][n];
	for (int i=n-1;i;i--){
		double sum=0.0;
		for (int j=i+1;j<=n;j++) sum+=ans[j]*a[i][j];
		ans[i]=(b[i]-sum)/a[i][i];
	}
}
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]);
	getEquation();
	gauss();
	for (int i=1;i<n;i++)
		printf("%.3lf ",ans[i]);
	printf("%.3lf\n",ans[n]);
	return 0;
}