矩阵元素查找

题目

描述
已知int一个有序矩阵mat，同时给定矩阵的大小n和m以及需要查找的元素x，且矩阵的行和列都是从小到大有序的。设计查找算法返回所查找元素的二元数组，代表该元素的行号和列号(均从零开始)。保证元素互异。
数据范围：0≤n,m≤1000
要求：空间复杂度 O(1)，时间复杂度 O(n+m)

思路

直接暴力遍历

代码

python版本：

无

c++版本：

class Solution {
public:
    vector<int> findElement(vector<vector<int> > mat, int n, int m, int x) {
        // write code here
        vector<int> res(2, 0);
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                if(mat[i][j]==x){
                    res[0] = i;
                    res[1] = j;
                }
            }
        }
        return res;
    }
};