行列排序矩阵找数

介绍

有一个矩阵，它的每一行的数都是按照从小到大排好序的，每一列也是从小到大排好序的
现在给出一个数，要求快速判断这个数在这个矩阵里面是否存在

思想

从第一行最后一列开始找
curR，curC表示当前位置的横纵坐标
有三种情况：
（1）arr[curR][curC] == num，直接返回true
（2）arr[curR][curC] > num，curC–
（3）arr[curR][curC] < num，curR++
直到走到最后一行的第一列还是没有找到该数，则返回false

查找函数find()

bool find(int **arr, int n, int m, int k){
    int curR = 0;
    int curC = m-1;
    while(curR < n && curC >= 0){
        if(arr[curR][curC] == k)
            return true;
        if(arr[curR][curC] > k){
            curC--;
        }else{
            curR++;
        }
    }
    return false;
}

相关阅读

旋转正方形矩阵
螺旋打印矩阵
Z字打印矩阵
行列排序矩阵找数

完整代码

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

int **inputArr(int n, int m){
    int **arr;
    arr = new int* [n];
    for(int i = 0; i < n; i++){
        arr[i] = new int [m];
    }
    for(int i = 0; i < n; i++){
        for(int j = 0; j < m; j++){
            cin >> arr[i][j];
        }
    }
    return arr;
}

bool find(int **arr, int n, int m, int k){
    int curR = 0;
    int curC = m-1;
    while(curR < n && curC >= 0){
        if(arr[curR][curC] == k)
            return true;
        if(arr[curR][curC] > k){
            curC--;
        }else{
            curR++;
        }
    }
    return false;
}

int main()
{
    int **arr;
    int n, m;
    cin >> n >> m;
    arr = inputArr(n, m);
    int k;
    cin >> k;
    cout << find(arr, n, m, k) << endl;
}