CD1 在行列都排好序的矩阵中找指定的数

CD1 在行列都排好序的矩阵中找指定的数

关键思路：简单模拟，即时比较

#include <bits/stdc++.h>
using namespace std;

const int N = 1e4 + 10;

int n, m, k;
int mrx[N][N];
bool flag = false;
int main () {
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            scanf ("%d", &mrx[i][j]);
            if (k == mrx[i][j]) {
                flag = true;
                break; // 及时跳出循环
            }
        }
    }
    
    if (flag) cout << "Yes" << endl;
    else cout << "No" << endl;
    
    return 0;
}

但是根据这个题目所说的，这个矩阵的每一行每一列是排好序的，那么我们就可以做一个优化，来减少时间；（根据行列升序的性质来进行优化）

#include <bits/stdc++.h>
using namespace std;

const int N = 1e4 + 10;

int n, m, k;
int mrx[N][N];
int main () {
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            scanf ("%d", &mrx[i][j]);
        }
    }
    
    // 从右上角开始，仔细想想，其实不会漏掉元素判断
    int i = 0;
    int j = m - 1;
    while (i < n &&j >= 0) {
        if (mrx[i][j] == k) {
            printf ("Yes\n");
            return 0;
        }
        else if (mrx[i][j] > k) {
            j--;
        }else{
            i++;
        }
    }
    
    printf ("No\n");
    
    
    return 0;
}

但我决定即时判断也可以算快的，能找到就可以及时退出了