Search a 2D Matrix

最新推荐文章于 2025-01-17 23:23:30 发布

原创最新推荐文章于 2025-01-17 23:23:30 发布 · 359 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

636 篇文章

订阅专栏

本文介绍了一种高效的算法，用于在一个特殊格式的二维矩阵中查找特定数值。该矩阵每一行的整数从左到右递增排序，并且每行的第一个整数大于前一行的最后一个整数。文章提供了一个具体的示例矩阵及目标值，并通过C++代码实现了解决方案。

Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:

Integers in each row are sorted from left to right.
The first integer of each row is greater than the last integer of the previous row.

For example,

Consider the following matrix:

[
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]

Given target = 3, return true.

class Solution {
public:
    bool searchMatrix(vector<vector<int> > &matrix, int target) {
        int size = matrix.size();
        if (size < 1)
        {
            return false;
        }

        int left = 0;
        int right = size - 1;
        while (left < right-1)
        {
            int mid = left + (right-left)/2;
            if (matrix[mid][0] == target)
            {
                return true;
            }
            else if (matrix[mid][0] > target)
            {
                right = mid - 1;
            }
            else
            {
                left = mid;
            }
        }

        if (matrix[left][0] == target || matrix[right][0] == target)
        {
            return true;
        }
        int row = left;
        if (left != right)
        {
            if (matrix[right][0] < target)
            {
                row = right;
            }
        }
    
        left = 0;
        right = matrix[row].size() - 1;
        while (left <= right)
        {
            int mid = left + (right-left)/2;
            if (matrix[row][mid] == target)
            {
                return true;
            }
            else if (matrix[row][mid] > target)
            {
                right = mid - 1;
            }
            else
            {
                left = mid + 1;
            }
        }

        return false;
    }
};