[剑指Offer] 4.二维数组中的查找(两种思路)

原创于 2020-04-07 19:54:34 发布 · 151 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

算法基础同时被 2 个专栏收录

21 篇文章

订阅专栏

剑指offer

13 篇文章

订阅专栏

本文介绍了一种在二维数组中查找目标值的高效算法。通过从右上角开始，根据目标值与当前元素的比较，逐步缩小查找范围，实现O(M+N)的时间复杂度。适用于按行、列递增排序的矩阵。

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

二维数组中的查找

解题思路

从右上角开始查找。矩阵中的一个数，它左边的数都比它小，下边的数都比它大。因此，从右上角开始查找，就可以根据 target 和当前元素的大小关系来缩小查找区间。

复杂度：O(M + N) + O(1)

public boolean Find(int target, int[][] matrix) {
    if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
        return false;
    int rows = matrix.length, cols = matrix[0].length;
    int r = 0, c = cols - 1; // 从右上角开始
    while (r <= rows - 1 && c >= 0) {
        if (target == matrix[r][c])
            return true;
        else if (target > matrix[r][c])
            r++;
        else
            c--;
    }
    return false;
}