九度 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\n");

        } else {

            printf("No\n");

        }

    }

    return 0;

}