高斯消元——模板

浮点数运算模板

题目链接

#include <bits/stdc++.h>
using namespace std;
const int N = 505;
const int M = 1005;
const double esp = 1e-6;
double a[N][M];
int n, m;

int gauss()
{
    //千万注意：所有判断精度的地方 fabs（）不要忘记否则会出现错误
    int c, r;
    for (c = 1, r = 1; c <= m; c ++)
    {
        int t = r;
        for (int i = r + 1; i <= n; i ++)           //找到最大行
        {
            if (fabs(a[i][c]) > fabs(a[t][c]))
                t = i;
        }

        if (fabs(a[t][c]) < esp) continue;                //如果最大行为 0 则不用计算

        for (int i = c; i <= m + 1; i ++) swap(a[t][i], a[r][i]);   //交换最大行

        for (int i = m + 1; i >= c; i --) a[r][i] /= a[r][c];       //把第 c 列的值变为 1

        for (int i = r + 1; i <= n; i ++)           //第 r 行下面的行中的第 c 列也变成 0
        {
            if (fabs(a[i][c]) > esp)
            {
                for (int j = m + 1; j >= c; j --)
                {
                    a[i][j] -= a[r][j] * a[i][c];
                }
            }
        }

        r ++;
    }


    if (r <= n)                             //这里必须是 <= 号，因为 r 是后缀++
    {
        for (int i = r; i <= n; i ++)
        {
            if (fabs(a[i][m + 1]) > esp)    //出现 （0 == 非0）情况，则无解
            {
                return 2;                   //无解
            }
        }
        return 1;                           //无穷解 有些行的出现（0 == 0）的情况, 则含有自由变量，怎有无数解
    }

    for (int i = n; i >= 1; i --)           //求解唯一解的过程
    {
        for (int j = i + 1; j <= m; j ++)
        {
            a[i][m + 1] -= a[i][j] * a[j][m + 1];
        }
    }

    return 0;                               //有唯一解
}



int main()
{
    scanf("%d", &n);                    //n行，m+1列, 第 m+1 列为当前行的方程的解
    m = n;
    for (int i = 1; i <= n; i ++)
        for (int j = 1; j <= m + 1; j ++)
            scanf("%lf", &a[i][j]);

    int f = gauss();
    if (f == 0)
    {
        for (int i = 1; i <= n; i ++)
        {
            printf("%.2f\n", a[i][m + 1]);
        }
    }
    else if (f == 1)
    {
        printf("Many solutions\n");
    }
    else if (f == 2)
    {
        printf("No solutions\n");
    }

    return 0;
}

异或方程组模板

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 110;

int n;
int a[N][N];
int gauss()
{
    int c,r;
    for(c=0,r=0;c<n;c++)
    {
        // 找主元
        int t=-1;
        for(int i=r;i<n;i++)
            if(a[i][c])
            {
                t=i;
                break;
            }
        if(t==-1) continue;
        // 交换主元行
        for(int j=c;j<=n;j++) swap(a[r][j],a[t][j]);
        // 左下角消
        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];
        r++;
    }
    // 判断
    if(r<n)
    {
        for(int i=r;i<n;i++)//i=r
            if(a[i][n])
                return 2;
        return 1;
    }
    // 消右上角
    for(int i=n-1;i>=0;i--)
        for(int j=i+1;j<n;j++)
            //如果是0 就不用下面的a[j][j] 来^a[i][j]了
            //如果不是0 才需要用第j行第j列a[j][j]来^第i行第j列a[i][j] 
            //进而进行整行row[i]^row[j] 间接导致 a[i][n]^a[j][n]
            if(a[i][j])
                a[i][n]^=a[j][n];
    return 0;
}

int main()
{
    cin >> n;
    for(int i=0;i<n;i++)
        for(int j=0;j<=n;j++)
            cin >> a[i][j];
    int t = gauss();
    if(t==0)
    {
        for(int i=0;i<n;i++) cout << a[i][n] << endl;
    }
    else if(t==1) puts("Multiple sets of solutions");
    else puts("No solution");
    return 0;
}

整数运算模板

#include<iostream>
#include<algorithm>
using namespace std;
int a[1005][1005];//增广矩阵
int x[505];//解集
bool free_x[505];//判断是否是不确定的变元
int free_num;
int equ,var;//equ个方程var个未知数，
//增广矩阵行数为equ,分别为0-equ-1;
//列数为var+1,分别为0-var
int gcd(int a,int b)
{
    if(b==0)
        return a;
    return gcd(b,a%b);
}

int lcm(int a,int b)
{
    return a/gcd(a,b)*b;
}

int gauss()
{
    int max_r,col,k,i,j;
    int temp,free_x_num,free_index;
    for(k=0,col=0;k<equ&&col<var;k++,col++)
    {
        max_r=k;
        for(i=k+1;i<equ;i++)
        {
            if(abs(a[i][col])>abs(a[max_r][col]))
                max_r=i;
        }
        if(a[max_r][col]==0)
        {
            k--;
            continue;
        }
        if(max_r!=k)
        {
            for(j=col;j<=var;j++)
                swap(a[k][j],a[max_r][j]);
        }
        for(i=k+1;i<equ;i++)
        {
            if(a[i][col]!=0)
            {
                int Lcm=lcm(abs(a[i][col]),abs(a[k][col]));
                int ta=Lcm/abs(a[i][col]);
                int tb=Lcm/abs(a[k][col]);
                if(a[i][col]*a[k][col]<0)
                    tb=-tb;
                for(j=col;j<=var;j++)
                {
                    a[i][j]=a[i][j]*ta-a[k][j]*tb;
                }
            }
        }
    }

    for(i=k;i<equ;i++)
        if(a[i][col]!=0)
            return -1;
    if(k<var)
    {
        //首先，自由变元有var-k个，即不确定的变元至少有var-k个
        for(i=k-1;i>=0;i--)
        {
            free_x_num=0;
            //用于判断该行中的不确定变元的个数，如果超过1个，则无法求解，它们仍然为不确定的变元
            for(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;
            temp=a[i][var];
            for(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(i=var-1;i>=0;i--)
    {
        int temp=a[i][var];
        for(j=i+1;j<var;j++)
        {
            temp-=a[i][j]*x[j];
        }
        x[i]=temp/a[i][i];
        if(temp%a[i][i]!=0)
            return -2;//有浮点数解
    }
    return 0;
}
int main()
{
    int n,m,i,j,ans;
    scanf("%d%d",&n,&m);
    equ=m;
    var=n;
    for(i=0;i<m;i++)
    {
        for(j=0;j<n;j++)
        {
            scanf("%d",&a[i][j]);
        }
        scanf("%d",&a[i][n]);
    }
    ans=gauss();
    if(ans==-1)
        printf("无解\n");
    else if(ans==-2)
        printf("有浮点数解\n");
    else if(ans>0)
    {
        printf("无穷多解，自由变元的个数 %d\n",ans);
        for(i=0;i<var;i++)
        {
            if(free_x[i])
                printf("不确定的 %d\n",i+1);
            else
                printf("%d %d\n",i+1,x[i]);
        }
    }
    else                            //有解
    {
        for(i=0;i<var;i++)
            printf("%d\n",x[i]);
    }
}