NC 矩阵元素查找

系列文章目录

---

---

前言

前些天发现了一个巨牛的人工智能学习网站，通俗易懂，风趣幽默，忍不住分享一下给大家。点击跳转到网站，这篇文章男女通用，看懂了就去分享给你的码吧。

---

描述
已知一个有序矩阵mat，同时给定矩阵的大小n和m以及需要查找的元素x，且矩阵的行和列都是从小到大有序的。设计查找算法返回所查找元素的二元数组，代表该元素的行号和列号(均从零开始)。保证元素互异。

class Solution {
public:
    vector<int> findElement(vector<vector<int> > mat, int n, int m, int x) {
        vector<int> res;
        if(n == 0)
            return res; 
        for(int i = n - 1, j = 0; i >= 0 && j < m; ){ //从最左下角的元素开始往左或往上
            if(mat[i][j] > x){   //元素较大，往上走
                i--;
            }
            else if(mat[i][j] < x){ //元素较小，往右走
                j++;
            }
            else{
                res.push_back(i);
                res.push_back(j);
                break;
            }
        }
        return res;
    }
};