2018.10.31 vijos1052贾老二算算术(高斯消元)
程序员文章站
2022-04-02 21:27:19
...
传送门
高斯消元模板题。
写的时候反了错误消元的时候除数和被除数反了。
所以把板子贴上来压压惊。
代码:
#include<bits/stdc++.h>
using namespace std;
inline int read(){
int ans=0,w=1;
char ch=getchar();
while(!isdigit(ch)){if(ch=='-')w=-1;ch=getchar();}
while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
return ans*w;
}
const int N=105;
int n;
double a[N][N],ans[N];
int main(){
n=read();
for(int i=1;i<=n;++i)for(int j=1;j<=n+1;++j)a[i][j]=read();
for(int i=1;i<=n;++i){
int tmp=i;
for(int j=i+1;j<=n;++j)if(fabs(a[j][i])>fabs(a[tmp][i]))tmp=j;
if(tmp^i)swap(a[tmp],a[i]);
for(int j=i+1;j<=n;++j){
double tmp=a[j][i]/a[i][i];
for(int k=1;k<=n+1;++k)a[j][k]-=a[i][k]*tmp;
}
}
for(int i=n;i;--i){
ans[i]=a[i][n+1]/a[i][i];
for(int j=i-1;j;--j)a[j][n+1]-=a[j][i]*ans[i];
}
for(int i=1;i<=n;++i)printf("%d ",(int)(ans[i]+0.5));
return 0;
}