Leetcode: Search a 2D Matrix

最新推荐文章于 2025-11-10 02:18:36 发布

原创最新推荐文章于 2025-11-10 02:18:36 发布 · 635 阅读

0 ·

CC 4.0 BY-SA版权

面试题专栏收录该内容

50 篇文章

订阅专栏

本文介绍了一种将二维矩阵视为有序列表并直接应用二分查找的方法。通过定义一个获取矩阵中特定索引值的函数，实现了对目标值的有效查找。

http://oj.leetcode.com/problems/search-a-2d-matrix/

Take the matrix as a sorted list, and then use binary search directly.

class Solution {
public:
    int GetValue(vector<vector<int> > &matrix, int index){
        int col=matrix[0].size();
        int c=index%col;
        int r=index/col;
        return matrix[r][c];
    }
    bool searchMatrix(vector<vector<int> > &matrix, int target) {
        if(matrix.size()==0) return false;
        int row=matrix.size(), col=matrix[0].size();
        int left=0, right=row*col-1;
        while(left<=right){
            int mid=(left+right)/2;
            int value=GetValue(matrix,mid);
            if(value==target) return true;
            else if(value<target) left=mid+1;
            else right=mid-1;
        }
        return false;
    }
};