74. Search a 2D Matrix

最新推荐文章于 2025-05-19 23:44:29 发布

weixin_30482181

最新推荐文章于 2025-05-19 23:44:29 发布

阅读量56

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/AndrewGhost/p/7211059.html

本文介绍了一种在二维矩阵中搜索特定元素的算法。该算法首先确定目标元素可能存在的行，然后对该行使用二分查找法来定位元素。适用于已按行、列排序的矩阵。

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m = matrix.size();
        if (!m)
        {
            return false;
        }
        int n = matrix.at(0).size();
        if (!n)
        {
            return false;
        }
        int row = -1;
        for (int i = 0; i < m; i++)
        {
            if (matrix[i][0] <= target && matrix[i][n - 1] >= target) {
                row = i;
                break;
            }
        }

        if (row < 0)
        {
            return false;
        }

        //二分查找
        int low = 0, high = n - 1;
        vector<int> tVec = matrix.at(row);
        while (low <= high)
        {
            int mid = (low + high) / 2;
            if (tVec.at(mid) > target)
            {
                high = mid - 1;
            }
            else if (tVec.at(mid) < target){
                low = mid + 1;
            }
            else {
                return true;
            }
        }

        return false;
    }
};