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

上下两个方程相减然后高斯消元

Description

Input

Output

Sample Input

Sample Output

HINT

Source

/* ***********************************************
Author        :CKboss
Created Time  :2015年05月15日 星期五 07时45分04秒
File Name     :BZOJ1013.cpp
************************************************ */

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstdlib>
#include <vector>
#include <queue>
#include <set>
#include <map>

using namespace std;

const double eps=1e-8;
const int maxn=40;

double a[maxn][maxn],x[maxn];
int equ,val;

bool Gauss()
{
	int i,j,k,col,max_r;
	for(k=0,col=0;k<equ&&col<val;k++,col++)
	{
		max_r=k;
		for(i=k+1;i<equ;i++)
			if(fabs(a[i][col])>fabs(a[max_r][col]))
				max_r=i;
		if(fabs(a[max_r][col])<eps)
			return false;
		if(k!=max_r)
		{
			for(int j=col;j<val;j++)
				swap(a[k][j],a[max_r][j]);
			swap(x[k],x[max_r]);
		}
		x[k]/=a[k][col];
		for(j=col+1;j<val;j++)
			a[k][j]/=a[k][col];
		a[k][col]=1;
		for(i=0;i<equ;i++)
		{
			if(i!=k)
			{
				x[i]-=x[k]*a[i][k];
				for(j=col+1;j<val;j++)
					a[i][j]-=a[k][j]*a[i][col];
				a[i][col]=0;
			}
		}
	}
	return true;
}

int n;
double cs[maxn][maxn];

int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);

	scanf("%d",&n);
	for(int i=0;i<n+1;i++)
		for(int j=0;j<n;j++)
			scanf("%lf",&cs[i][j]);

	// build matrix
	equ=n; val=n;
	for(int i=0;i<n;i++)
	{
		int ii=i+1;
		double sum=0;
		for(int j=0;j<n;j++)
		{
			a[i][j]=2*(cs[ii][j]-cs[i][j]);
			sum+=(cs[ii][j]-cs[i][j])*(cs[ii][j]+cs[i][j]);
		}
		x[i]=sum;
	}

	Gauss();

	for(int i=0;i<val;i++)
	{
		printf("%.3lf%c",x[i],(i==val-1)?'\n':' ');
	}
    
    return 0;
}