第一道高斯消元。嗯。特判啥的还都没写。待改进。
题意比较简单,n+1个点,可以列出n个方程,所以就是要解一个线性方程组。我们用高斯消元来解。首先线性方程组可以表示成A.X=C的形式,其中A为系数矩阵,X为未知数矩阵(一个列向量),C为常数矩阵。现在我们已知A和C,求解X。大概就是加减消元的思想。时间复杂度
O(n3)
.
讲解:https://www.cnblogs.com/candy99/p/6653743.html
http://www.cppblog.com/menjitianya/archive/2014/06/08/207226.html
例子:https://jingyan.baidu.com/article/39810a23e40c80b636fda63a.html
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define inf 0x3f3f3f3f
#define N 20
#define eps 1e-6
#define ll long long
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int n;
double a[N][N],f[N];
inline bool Gauss(){
for(int i=1;i<=n;++i){
int r=i;
for(int j=i+1;j<=n;++j) if(fabs(a[j][i])>fabs(a[r][i])) r=j;
if(fabs(a[r][i])<eps) continue;
if(r!=i) for(int j=1;j<=n+1;++j) swap(a[i][j],a[r][j]);
for(int j=i+1;j<=n;++j){
double t=a[j][i]/a[i][i];
for(int k=i;k<=n+1;++k) a[j][k]-=t*a[i][k];
}
}for(int i=n;i>=1;--i){
for(int j=n;j>i;--j) a[i][n+1]-=a[i][j]*a[j][n+1];
a[i][n+1]/=a[i][i];
}return 1;
}
int main(){
// freopen("a.in","r",stdin);
n=read();
for(int i=1;i<=n;++i) scanf("%lf",&f[i]);
for(int i=1;i<=n;++i)
for(int j=1;j<=n;++j){
double x;scanf("%lf",&x);
a[i][j]=(x-f[j])*2;a[i][n+1]+=x*x-f[j]*f[j];
}Gauss();
printf("%.3lf",a[1][n+1]);
for(int i=2;i<=n;++i) printf(" %.3lf",a[i][n+1]);
return 0;
}