BZOJ 1013(Gauss消元)

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 )

思路

表面的点到球心的距离是相等的

考虑有n+1个方程 n+1个未知数（球心和半径）

但是这样会出现二次 无法求解

考虑先将r消去 同时可以消去二次的未知数

从而解齐次线性方程组

普通的高斯消元

代码示例

/**************************************************************
    Problem: 1013
    User: Feynman1999
    Language: C++
    Result: Accepted
    Time:40 ms
    Memory:1300 kb
****************************************************************/

//高斯消元 时间复杂度O(n^3) 使用浮点数计算
#include<bits/stdc++.h>
using namespace std;
const double eps=1e-8;
const int maxn=101;
double det;//行列式的值

//也可以避免实数运算 使用辗转相减的方法 多一个log的时间复杂度
//特别适合于行列式求值 取模操作 (除法要求逆元)
inline int solve(double a[][maxn],bool l[],double ans[],const int &n,const int &m){
    //a为方程组对应的矩阵
    //l,ans存储解  l[]表示是否为自由变元 1表示不是 0表示是
    //n为未知数的个数 m为方程的个数
    //如果无解返回-1 否则返回自由变元数
    int res=0,r=0;//r为第几行 res为自由变元数
    for(int i=0;i<n;++i) l[i]=false;//开始都是自由变元

    for(int i=0;i<n;++i){//枚举列

        for(int j=r;j<m;++j)//枚举行
            if(fabs(a[j][i])>eps){
                //找到当前列下从第r行开始第一个不为零的元素并交换到第r行
                //如果一直为0 则下面会continue res++ 即自由变元+1
                //如果有不为0的 把它调上来（交换行）
                for(int k=i;k<=n;++k)//第j行和第r行交换 因为a[j][i]!=0
                    swap(a[j][k],a[r][k]);
                break;
            }

        //从a[r][i]这一个元素开始 往下的每个元素都是0了 所以这个元素xi是自由变元 i+1跳过即可
        if(fabs(a[r][i])<eps){
            ++res;
            continue;
        }

        for(int j=0;j<m;++j)//j是行数 从第一行（j=0）开始  让上三角更简洁 这样后面求ans就没有必要从下往上了
            if(j!=r && fabs(a[j][i])>eps){
                double tmp=a[j][i]/a[r][i];
                for(int k=i;k<=n;++k)
                    a[j][k]-=tmp*a[r][k];
            }
        l[i]=true,++r;
    }

    det=1;
    for(int i=0;i<n;++i)
        det*=a[i][i];

    //检查是否无解
    for(int i=n-res;i<m;++i){
        if(fabs(a[i][n])>eps) return -1;
    }

    //输出阶梯型矩阵
//    for(int j=0;j<m;++j){
//        for(int k=0;k<=n;++k){
//            cout<<a[j][k]<<' ';
//        }
//        cout<<endl;
//    }


    //下面求结果
    for(int i=0;i<n;++i){
        if(l[i])//不是自由变元
            for(int j=0;j<n;++j)
                if(fabs(a[j][i])>eps)
                        ans[i]=a[j][n]/a[j][i];
    }
    return res;//返回自由变元数
}

int main()
{
    //ios::sync_with_stdio(false);
    //freopen("read.txt","r",stdin);
    //freopen("out1.txt","w",stdout);
    double a[maxn][maxn];
    bool l[maxn];
    double ans[maxn];
    double arr[maxn];
    double brr[maxn];
    int n,m,res;
//    int t;
//    scanf("%d",&t);
//    while(t--)
//    {
        memset(a,0,sizeof(a));
        scanf("%d",&n);
        m=n;
        for(int i=0;i<n;++i) scanf("%lf",&arr[i]);
        for(int i=0;i<n;++i){
            for(int j=0;j<n;++j) scanf("%lf",&brr[j]);
            for(int j=0;j<n;++j) a[i][j]=2*(brr[j]-arr[j]);
            for(int j=0;j<n;++j) a[i][n]+=(brr[j]*brr[j]-arr[j]*arr[j]);
            for(int j=0;j<n;++j) arr[j]=brr[j];
        }
        res=solve(a,l,ans,n,m);
        for(int i=0;i<n;++i)
            if(i==n-1) printf("%.3lf\n",ans[i]);
            else printf("%.3lf ",ans[i]);
//    }
    return 0;
}