每日一题：leetcode74.搜索二维矩阵

本文链接：https://blog.csdn.net/T_T233333333/article/details/115320050

题目描述

在这里插入图片描述

题目分析

感觉这是一个放错标签的简单题。题目非常简单，思路应该很明确是二分，我很快写了一个（虽然不小心把!=打成==调试了一会）。

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        if (target < matrix[0][0] || target > matrix.back().back()) return false;
        auto result = row_lower_bound(matrix, 0, matrix.size(), target);
        if (result.first) {
            int x = result.second;
            const auto &vec = matrix[x];
            return  lower_bound(vec.cbegin(), vec.cend(), target) != upper_bound(vec.cbegin(), vec.cend(), target);
        } else {
            return false;
        }

    }
private:
    pair<bool, int> row_lower_bound(vector<vector<int>> &matrix, int l, int r, int target) {;
        while (l < r) {
            int mid = (l + r) >> 1;
            if (target >= matrix[mid][0] && target <= matrix[mid].back()) return {true, mid};
            else if (target < matrix[mid][0]) r = mid;
            else l = mid + 1;
        }
        return {false, 0};
    }
};

但是当我看到题解的时候我觉得我对STL一无所知。虽然已经学习了lambda表达式，但是不太能够灵活运用。对于这种STL已经实现了的算法，有些细节可能不同，我们需要传递一个lambda表达式定制一下操作就可以了。

class Solution {
public:
    bool searchMatrix(vector<vector<int>> matrix, int target) {
        auto row = upper_bound(matrix.begin(), matrix.end(), target, [](const int b, const vector<int> &a) {
            return b < a[0];
        });
        if (row == matrix.begin()) {
            return false;
        }
        --row;
        return binary_search(row->begin(), row->end(), target);
    }
};