ZOJ 3645 （简单的高斯消元）

题目链接： http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3645

题意：给你12个方程，让你解出来11个变量的值。

思路： 此题目的关键是如果出来二次项，仔细观察会发现每一个方程都包含一个sigma(Xi^2) (1 <= i <= 11)的项，所以我们可以把这个项整体当成一个未知数，那么整个题目就是12个未知数，12个方程了，再直接套用高斯消元的模板即可。

code：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const double eps = 1e-6;
typedef vector<double> vec;
typedef vector<vec> mat;

vec gauss_jordan(const mat& A,const vec& b)
{
	int n=A.size();
	mat B(n,vec(n+1));
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++)B[i][j]=A[i][j];
	}
	for(int i=0;i<n;i++)B[i][n]=b[i];
	for(int i=0;i<n;i++){
		int pivot = i;
		for(int j=i;j<n;j++){
			if(abs(B[j][i])>abs(B[pivot][i]))pivot=j;
		}
		swap(B[i],B[pivot]);
		if(abs(B[i][i])<eps) return vec();
		for(int j=i+1;j<=n;j++)B[i][j]/=B[i][i];
		for(int j=0;j<n;j++){
			if(i!=j){
				for(int k=i+1;k<=n;k++){
					B[j][k]-=B[j][i]*B[i][k];
				}
			}
		}
	}
	vec x(n);
	for(int i=0;i<n;i++)x[i]=B[i][n];
	return x;
}
double c[13][12], d[13];
int main()
{
	int T; cin>>T;
	while(T--){
		for(int i=0; i<12; i++){
                        for(int j=0; j<11; j++){
				cin>>c[i][j];
                        }
                        cin>>d[i];
		}
		mat A(11,vec(11));
		vec v(11);
		for(int i=0; i<11; i++){
			v[i]=0;
			for(int j=0; j<11; j++){
				A[i][j]=(-c[i][j]+c[i+1][j])*2.0;
				v[i]-=(c[i][j]*c[i][j]-c[i+1][j]*c[i+1][j]);
			}
			v[i]+=d[i]*d[i]-d[i+1]*d[i+1];
		}
		vec ans=gauss_jordan(A,v);
		for(int i=0; i<11; i++){
			double x=ans[i];
			if(abs(x)<eps) x=0;
			printf("%.2f%c",x,i==10?'\n':' ');
		}
	}
}