高斯消元模板

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
const int maxn=100;
int equ,var;
int a[maxn][maxn];
int x[maxn];
bool free_x[maxn];
int free_num;
//高斯消元，-2表示有浮点数解，-1表示无解，0表示唯一解，大于0表示无穷解
int Gauss()
{
    int max_r,ta,tb,LCM,free_x_num,free_index;
    int col=0,k;
    for(k=0;k<equ&&col<var;k++,col++)
    {
        max_r=k;
        for(int i=k+1;i<equ;i++)
            if(abs(a[i][col])>abs(a[max_r][col]))max_r=i;
        if(max_r!=k)
            for(int i=k;i<=var;i++)swap(a[k][i],a[max_r][i]);
        if(a[k][col]==0)
        {//说明该col列以下全是0，则处理当前行的下一列
            k--;
            continue;
        }
        for(int i=k+1;i<equ;i++)
        {
            if(a[i][col]!=0)
            {
                LCM=lcm(abs(a[i][col]),abs(a[k][col]));
                ta=LCM/abs(a[i][col]),tb=LCM/abs(a[k][col]);
                if(a[i][col]*a[k][col]<0)tb=-tb;//异号的情况是两个数相加
                for(int j=col;j<=var;j++)
                    a[i][j]=a[i][j]*ta-a[k][j]*tb;
            }
        }
    }
    //无解的情况存在(0,0,...,a)
    for(int i=k;i<equ;i++)
        if(a[i][col]!=0)return -1;
    //无穷解的情况，出现全0的行
    if(k<var)
    {
        //自由变量有var-k个，即不确定的变量至少有var-k个
        for(int i=k-1;i>=0;i--)
        {
            // 第i行一定不会是(0, 0, ..., 0)的情况，因为这样的行是在第k行到第equ行.
            // 同样，第i行一定不会是(0, 0, ..., a), a != 0的情况，这样的无解的.
            free_x_num=0;// 用于判断该行中的不确定的变元的个数，如果超过1个，则无法求解，它们仍然为不确定的变元.
            for(int j=0;j<var;j++)
                if(a[i][j]!=0&&free_x[j])free_x_num++,free_index=j;
            if(free_x_num>1)continue;
            // 说明就只有一个不确定的变元free_index，那么可以求解出该变元，且该变元是确定的.
            temp=a[i][var];
            for(int j=0;j<var;j++)
                if(a[i][j]!=0&&j!=free_index)temp-=a[i][j]*x[j];
            x[free_index]=temp/a[i][free_index];//求出该便变元
            free_x[free_index]=0;
        }
        return var-k;
    }
    //唯一解
    for(int i=var-1;i>=0;i--)
    {
        temp=a[i][var];
        for(int j=i+1;j<var;j++)
            if(a[i][j]!=0)temp-=a[i][j]*x[j];
        if(temp%a[i][i]!=0)return -2;
        x[i]=temp/a[i][i];
    }
    return 0;
}