二维数组的查找

最新推荐文章于 2024-03-30 17:52:23 发布

原创最新推荐文章于 2024-03-30 17:52:23 发布 · 412 阅读

1 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种高效的查找算法，用于在一个NxM的整数矩阵中查找特定元素x的位置，矩阵的行和列均按升序排列。算法通过比较目标值与矩阵右上角元素来逐步缩小搜索范围，直至找到目标值或确认不存在。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

有一个NxM的整数矩阵，矩阵的行和列都是从小到大有序的。请设计一个高效的查找算法，查找矩阵中元素x的位置。

给定一个int有序矩阵mat，同时给定矩阵的大小n和m以及需要查找的元素x，请返回一个二元数组，代表该元素的行号和列号(均从零开始)。保证元素互异。

测试样例：

[[1,2,3],[4,5,6]],2,3,6

返回：[1,2]

代码：

class Finder {
public:
    vector<int> findElement(vector<vector<int> > mat, int n, int m, int x) {
        // write code here
        int i=0;
        int j=m-1;
        vector<int> result;
        while(i<n&&j>=0){
            if(x==mat[i][j]) 
                break;            
            if(x>mat[i][j])
                i++;
            else
                j--;                    
        }
        
        result.push_back(i);
        result.push_back(j);
        return result;
    }
};