九度 1384 二维数组中的查找

#include<cstdio>
#include<iostream>
using namespace std;
#define NMAX 1000010
int arr[NMAX];
/**
 * 只需要从左下角或右上角开始查找，即可。
 * 当是从右上角开始查找时，
 * 左上角（row = 0,column = columns - 1），
 * 1）若查找的值key大于当前的值arr[i]，则说明key在此行以下，故有row++；
 * 2)若查找的值key小于当前的值arr[i]，则说明key在此列的左边，故有column--；
 * 3）若相等，则返回true
 * 4）判断条件：row<rows && column >=0，若满足，继续1），若不满足，则直接返回false
 */
bool query(int rows, int columns, int key) {
    bool flag = false;
    int column = columns - 1;
    int row = 0;
    while (row < rows && column >= 0) {
        int temp = arr[row * columns + column];
        if (temp == key) {
            flag = true;
            return flag;
        } else if (temp > key) {
            column--;
        } else {
            row++;
        }
    }
    return flag;
}
/**
 * 当是从左下角开始查找时，
 * 左上角（row = rows - 1,column = 0），
 * 1）若查找的值key小于当前的值arr[i]，则说明key在此行以下，故有row--；
 * 2)若查找的值key大于当前的值arr[i]，则说明key在此列的左边，故有column++；
 * 3）若相等，则返回true
 * 4）判断条件：row >= 0 && column < columns，若满足，继续1），若不满足，则直接返回false
 */
bool query1(int rows, int columns, int key) {
    bool flag = false;
    int column = 0;
    int row = rows - 1;
    int temp = -1;
    while (row >= 0 && column < columns) {
        temp = arr[row * columns + column];
        if (temp == key) {
            flag = true;
            return flag;
        } else if (temp > key) {
            row--;
        } else {
            column++;
        }
    }
    return false;
}
int main() {
    int n, m, t, len;
    bool flag;
    while (~scanf("%d%d%d", &n, &m,&t)) {
        len = n * m;
        for (int i = 0; i < len; i++) {
            scanf("%d", arr + i);
        }
        flag = query1(n, m, t);
        if (flag) {
            printf("Yes
");
        } else {
            printf("No
");
        }
    }
    return 0;
}