蓝桥杯算法训练-JOE的矩阵

问题描述

问题描述
　　最近JOE又在线性代数的模拟考中拿满分了，这直接导致了JOE对于计算矩阵的热情急剧下降，所以JOE希望能有这样一个程序能帮助他计算矩阵的秩。
输入格式
　　第一行，两个数n,m，表示矩阵是n*m的。
　　下面共n行，每行m个数（可能为负），表示这个矩阵。
输出格式
　　这个矩阵的秩。
样例输入
2 3
1 2 1
2 4 3
样例输出
1

代码

就是求矩阵的秩，使用高斯消元即可解决。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>

using namespace std;

const int N = 110;
const double eps = 1e-8;

double a[N][N];
int n, m;

int gauss() {
    int r, c;
    for (r = 0, c = 0; c < n; c++) {

        // 1. 找到最大值
        int t = r;
        for (int i = r; i < n; i++) {
            if (fabs(a[i][c]) > fabs(a[t][c]))
                t = i;
        }

        if (fabs(a[t][c]) < eps) continue;

        // 2. 交换到首行
        for (int i = c; i < m; i++) swap(a[r][i], a[t][i]);

        // 3. 本行首列置为1
        for (int i = m - 1; i >= c; i--) a[r][i] /= a[r][c];

        // 4. 下面的行首列置为0
        for (int i = r + 1; i < n; i++) {
            if (fabs(a[i][c]) < eps) continue;
            for (int j = m - 1; j >= c; j--) {
                a[i][j] -= a[i][c] * a[r][j]; // - 首航本列 * i, r首列商
            }
        }

        r++;
    }
    return r;
}

int main() {
    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            scanf("%lf", &a[i][j]);
            // printf("%.2lf\n", a[i][j]);
        }
    }

    int t = gauss();
    printf("%d\n", t);
    return 0;
}