高斯消元:
难点就是将思路以代码的形式展开:
题目:883. 高斯消元解线性方程组 - AcWing题库
思路:
①先将所要消元的行中,首列元素绝对值最大的那一行找出来
②与所要消元的行中的首行交换位置,③将首行的第一个元素化为1(那么首行的其余元素也需要改变)④将首行下面的第一列全部调为0
举例:
当然第三次遍历时,①、②还是原式, 第③步时,第三行变为:0 0 1 -1/2,第④不变。然后跳出
第一层for(c=0,r=0;c<n;c++)循环,开始向前求解
代码:
/*
高斯消元可以在n^3的时间内,求解n个方程n个未知量的线性方程组
*/
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
const int N=110;
//防止精度问题
const double esp=1e-8;
int n;
double a[N][N];
//debug时可调用此函数来检验
/*
void out()
{
for(int i=0;i<n;i++)
{
for(int j=0;j<n+1;j++)
printf("%.2lf ",a[i][j]);
cout << endl;
}
cout << endl;
}
*/
int guss()
{
//c表示枚举的是哪一列,r表示枚举的是哪一行//①
int c,r;
//从第一行第一列开始逐渐向后求。即首先化第一列除首行的所有值为0,在化第二列除第二行的所有值为0......//②
for(c=0,r=0;c<n;c++)
{
//out ();
//记录当前在消倒三角时,不需要更改值的行的最顶层的那一行//③
int t=r;
for(int i=r;i<n;i++)
if(fabs(a[i][c])>fabs(a[t][c]))
//找到除了已经确定了的行之外,剩下所有行中第一个非0列里值最大的那一行//④
//并记录下它所在的行//
t=i;
//如果该点为0,则说明该点以及该点以下的所有点全为0,跳过即可//⑤
if(fabs(a[t][c]) < esp) continue;
//将所需要交换值的那一行(t)的所有值,与不需要更改值的最顶层那一行交换位置//⑥
for(int j=c;j<n+1;j++) swap(a[r][j],a[t][j]);
//将该行首列化为1,注意应该从后向前化,省去记录首列元素值的代码//⑦
for(int j=n;j>=c;j--) a[r][j]/=a[r][c];
//遍历更新,将第c列,除第r行的所有值化为0。在这个过程中,消0的点所在的行也需要改变值//⑧
for(int i=r+1;i<n;i++)
//如果该点已经为0,则跳过。只计算非0的值
if(fabs(a[i][c]) > esp)
for(int j=n;j>=c;j--)
a[i][j]-=a[i][c]*a[r][j];
//从最后一行向前化0也可以:
/*
for(int i=n-1;i>r;i--)
//等于0时跳过就行了
if(fabs(a[i][c]) > esp)
for(int j=n;j>=c;j--)
a[i][j]-=a[r][j]*a[i][c];
*/
//固定第r行,说明第r行及其以上全为倒三角型,r向下移动,将下面的化为倒三角型//⑨
r++;
}
if(r < n)//说明有系数矩阵的行或者有矩阵的行全为0
{
for(int i=r;i<n;i++)
if(fabs(a[i][n]) > esp)//如果有系数矩阵不全为0,那么r必然=n,与if矛盾,则无解⑩
return 2;
return 1;//说明有行全为0,则为无穷多解//11
}
//这里建议手动模拟一下
for(int i=n-1;i>=0;i--)
for(int j=i+1;j<n;j++)
a[i][n]=a[i][n]-a[j][n]*a[i][j];//逐渐向前求解,并将值赋给a[i][n]而非a[i][i]来简化代码
return 0;
}
int main()
{
cin >> n;
//读入数据,注意需要将增广矩阵的值也读入进来
for(int i=0;i<n;i++)
for(int j=0;j<n+1;j++)
cin >> a[i][j];
int t=guss();
if(t==0)
{
//将所有答案记录在第n列即可
for(int i=0;i<n;i++) printf("%.2lf\n",a[i][n]);
}
else if(t==1) puts("Infinite group solutions");
else puts("No solution");
return 0;
}
贴一张上述不同状态的图:高斯消元不同状态贴图 - 知乎 (zhihu.com)
题目:884. 高斯消元解异或线性方程组 - AcWing题库
代码:
普通的高斯消元消成倒三角时,通常采用四则运算,本题采用抑或。其余思路不变
代码:
//抑或 ^ :相同为0,不同为1。(不进位加法)
#include<iostream>
#include<algorithm>
using namespace std;
const int N=105;
int n,a[N][N];
int guass()
{
int c,r;
for(c = 0, r = 0; c < n; c++)
{
int t=r;
for(int i=r;i<n;i ++ )
if(a[i][c])
t=i;
//如果挑出来的首列仍为0的话,则证明该列全为0,直接跳过该次循环即可
if(!a[t][c]) continue;
for(int i=c;i<n+1;i++) swap(a[r][i], a[t][i]);
//从下一行的第c列开始逐渐向后更新第c列
for(int i=r+1;i<n;i++)
if(a[i][c])
for(int j=n;j>=c;j--)
a[i][j] ^=a[r][j];//更新第i行的所有值
r++;
}
if(r < n)
{
for(int i=r;i<n;i++)
if(a[i][n])//存在诸如1=0的式子表示无解
return 1;
return 2;
}
for(int i=n-1;i>=0;i--)
for(int j=i+1;j<n;j++)
a[i][n]^=a[i][j]*a[j][n];//当前这一行的系数*下一行x的值
return 0;
}
int main()
{
cin >> n;
for(int i = 0; i < n; i ++ )
for(int j = 0; j < n+1; j ++ )
cin >> a[i][j];
int t=guass();
if(t==1) cout << "No solution" << endl;
else if (t == 2) cout << "Multiple sets of solutions" << endl;
else
{
for(int i=0;i<n;i++) cout << a[i][n] << endl;
}
return 0;
}