式子列一下,发现二次项被消掉,于是就是一个n元一次方程组。
第一次写Guass消元,刚开始压力巨大,开始对着红书撸代码。
结果写着写着觉得书上的代码好繁琐,既然思路懂了,直接就把书扔一边了。
直接重写!由于这题没有出现无解的情况,所以代码写起来十分舒畅。
然后过了样例,然后就过了,今天感觉真不错。
-------------------------------------------------------------------------------------------------------------------------------------
Code:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef double ld;
const int N=15;
const ld eps=1e-12;
ld p[2][N],sum1,sum2;
ld s[N][N],ans[N];
int n;
void solve(){
for(int i=1;i<=n;i++){
for(int j=i;j<=n;j++)
if(fabs(s[j][i])>eps){
for(int k=1;k<=n+1;k++) swap(s[i][k],s[j][k]);
break;
}
for(int j=i+1;j<=n;j++){
ld tmp=s[j][i]/s[i][i];
for(int k=1;k<=n+1;k++) s[j][k]-=tmp*s[i][k];
}
}for(int i=n;i>=1;i--){
for(int k=i+1;k<=n;k++){
s[i][n+1]-=s[i][k]*ans[k];
s[i][k]=0;
}ans[i]=s[i][n+1]/s[i][i];
}
}
int main(){
int flag=0;
scanf("%d",&n);
for(int i=1;i<=n+1;i++){
flag^=1;
memset(p[flag],0,sizeof(p[flag]));
for(int j=1;j<=n;j++){
scanf("%lf",&p[flag][j]);
p[flag][n+1]+=p[flag][j]*p[flag][j];
}if (i==1) continue;
for(int j=1;j<=n;j++)
s[i-1][j]=2*(p[flag^1][j]-p[flag][j]);
s[i-1][n+1]=-p[flag][n+1]+p[flag^1][n+1];
}solve();
for(int i=1;i<=n-1;i++) printf("%.3lf ",ans[i]);
printf("%.3lf",ans[n]);
return 0;
}