HihoCoder 1195 高斯消元·一(高斯消元)

题意

https://hihocoder.com/problemset/problem/1195

思路

高斯消元是解决高元方程的一种算法,复杂度 \(O(n^3)\) 。

过程大致是:

  1. 构造一个未知数的倒三角,并维护多解标记;
  2. 寻找是否出现没有未知数但常数非零的式子,有则返回无解;
  3. 多解标记若存在则返回多解;
  4. 在倒三角里倒着扫一遍,解出所有未知数。

下面是代码实现:

代码

#include<bits/stdc++.h>
#define FOR(i,x,y) for(int i=(x),i##END=(y);i<=i##END;++i)
#define DOR(i,x,y) for(int i=(x),i##END=(y);i>=i##END;--i)
typedef long long LL;
using namespace std;
const int N=505;
const double eps=1e-8;
double a[2*N][N],b[2*N];
int n,m; int Gauss(double a[2*N][N],double b[2*N],int n,int m)
{
bool flag=0;
for(int i=1,r=1;i<=n;i++,r++)
{
bool f=0;
FOR(j,r,m)if(fabs(a[j][i])>eps)
{
swap(a[j],a[r]),swap(b[j],b[r]);
f=1;break;
}
if(!f){flag=1,r--;continue;}
FOR(j,r+1,m)
{
FOR(k,i+1,n)a[j][k]-=a[r][k]*a[j][i]/a[r][i];
b[j]-=b[r]*a[j][i]/a[r][i];
a[j][i]=0;
}
} FOR(i,1,m)if(fabs(b[i])>eps)
{
bool f=0;
FOR(j,1,n)if(fabs(a[i][j])>eps){f=1;break;}
if(!f)return 0;
}
if(flag)return -1; DOR(i,n,1)
{
FOR(j,i+1,n)b[i]-=a[i][j]*b[j];
b[i]/=a[i][i];
a[i][i]=1;
}
return 1;
} int main()
{
scanf("%d%d",&n,&m);
FOR(i,1,m)
{
FOR(j,1,n)scanf("%lf",&a[i][j]);
scanf("%lf",&b[i]);
}
int res=Gauss(a,b,n,m);
if(res==-1)puts("Many solutions");
else if(res==0)puts("No solutions");
else FOR(i,1,n)printf("%d\n",(int)(b[i]+0.5));
return 0;
}
上一篇:【Android开发日记】之基础篇(二)——Android的动画效果


下一篇:Entity Framework Core系列之DbContext(删除)