文章标题 高斯消元解方程组（模板）

参考自http://www.cnblogs.com/kuangbin/archive/2012/09/01/2667044.html

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <queue>
#include <set>
#include <map>
#include <algorithm>
#include <math.h>
#include <vector>
using namespace std;
typedef long long ll;

const int mod=1e9+7;
const int maxn=105;

ll mat[maxn][maxn];//增广矩阵 
ll x[maxn];//解集 
bool free_x[maxn];//标记是否是自由变元 

ll lcm(ll a,ll b){
    return a/__gcd(a,b)*b;
} 

// 高斯消元法解方程组(Gauss-Jordan elimination).(-2表示有浮点数解，但无整数解，
//-1表示无解，0表示唯一解，大于0表示无穷解，并返回自由变元的个数)
//有equ个方程，var个变元。增广矩阵行数为equ,分别为0到equ-1,列数为var+1,分别为0到var.
int Guass(int equ,int var){
    int i,j,k;
    int max_r;// 当前这列绝对值最大的行.
    int col;//当前处理的列
    ll ta,tb,LCM;
    int free_num;
    int free_index;
    memset (x,0,sizeof (x));
    memset (free_x,true,sizeof (free_x));
    for (k=0,col=0;k<equ&&col<var;k++,col++){
        max_r=k;
        for (int i=k+1;i<equ;i++){
            if (fabs(mat[i][col])>fabs(mat[max_r][col]))max_r=i;
        }
        if (max_r!=k){//交换 
            for (int i=k;i<var+1;i++)swap(mat[max_r][i],mat[k][i]);
        }
        if(mat[k][col]==0){// 说明该col列第k行以下全是0了，则处理当前行的下一列.
            k--;
            continue;
        } 
        for (int i=k+1;i<equ;i++)if (mat[i][col]!=0) {//将k后边的col进行初等变换成行阶梯矩阵
            LCM=lcm(mat[i][col],mat[k][col]);
            ta=LCM/mat[i][col];
            tb=LCM/mat[k][col];
            if (mat[i][col]*mat[k][col]<0)tb=-tb;//异号的情况是相加
            for (int j=col;j<var+1;j++){//第i行先乘ta倍-第k行乘tb倍 
                mat[i][j]=mat[i][j]*ta-tb*mat[k][j];
            }
        }
    }
    // 1. 无解的情况: 化简的增广阵中存在(0, 0, ..., a)这样的行(a != 0).
    for (int i=k;i<equ;i++){
        if (mat[i][col]!=0)return -1;
    }
    // 2. 无穷解的情况: 在var * (var + 1)的增广阵中出现(0, 0, ..., 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_num=0;
            for (int j=0;j<var;j++)if (mat[i][j]!=0&&free_x[j]){
                free_num++;
                free_index=j;
            }
            if (free_num>1){
                continue;// 无法求解出确定的变元.
            }
            ll tmp=mat[i][var];
            for (int j=0;j<var;j++)if (mat[i][j]!=0&&j!=free_index){
                tmp=tmp-mat[i][j]*x[j];
            }
            x[free_index]=tmp/mat[i][free_index];//求出该变元
            free_x[free_index]=false;// 该变元是确定的.
        }
        return var-k;// 自由变元有var - k个. 
    }
    // 3. 唯一解的情况: 在var * (var + 1)的增广阵中形成严格的上三角阵.
    // 计算出Xn-1, Xn-2 ... X0.
    for (int i=var-1;i>=0;i--){
        ll tmp=mat[i][var];
        for (int j=i+1;j<var;j++){
            tmp=tmp-mat[i][j]*x[j];
        }
        if (tmp%mat[i][i])return -2;// 说明有浮点数解，但无整数解.
        x[i]=tmp/mat[i][i];
    }
    return 0;
}

int main()
{

    return 0;
}