bzoj1013: [JSOI2008]球形空间产生器sphere[Guass消元]

来源:互联网 发布:ubuntu和kali 编辑:程序博客网 时间:2024/05/16 23:59

式子列一下,发现二次项被消掉,于是就是一个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;} 


0 0
原创粉丝点击