码迷,mamicode.com
首页 > 编程语言 > 详细

【剑指Offer】 04 二维数组中的查找

时间:2021-01-14 10:57:46      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:iterator   排序   cto   while   二维   有序二叉树   括号   就是   顺序   

题目

在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

示例:

现有矩阵 matrix 如下:

[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。给定 target = 20,返回 false。

限制:

0 <= n <= 1000, 0 <= m <= 1000

思考

  • 一看到 我就十分暴力的想从左上角开始查找
  • 但是写出来的就是一个巨大的bug
  • 有很多没有考虑到的情况
  • 然后想二分,但是又不会写
  • 看到题解有人说可以看成一个有序二叉树!!!突然get到直接从右上角查找即可
    • 如果target小于当前节点,则查左子树,即col--;
    • 如果target大于当前节点,则查找右子树,即row++
    • 如果target等于当前节点,则return true
    • 边界判断,如果越界则return false
  • 关于vector
    • 一开始用c++写的时候,对vector的操作很陌生,不知道怎么简单操作下标的移动,一开始就只知道用iterator,然后发现可以用size()
  • java实现
    • 然后用Java实现的时候,发现我基本上对Java一无所知
      • 输入是二维数组,下标可以直接用符号。
      • 长度的话,可以用length成员,不用括号!!
      • 不能用‘!’来看是不是0?

代码

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        int n = matrix.length;
        if(n==0) return false;
        int m = matrix[0].length;
        if(m==0) return false;
        int row = 0, col = m-1;
        while(row<=n-1 && col>=0){
            if(target==matrix[row][col]) return true;
            else if(target<matrix[row][col]) col--;
            else if(target>matrix[row][col]) row++;
        }
        return false;
    }
}

【剑指Offer】 04 二维数组中的查找

标签:iterator   排序   cto   while   二维   有序二叉树   括号   就是   顺序   

原文地址:https://www.cnblogs.com/xuwanwei/p/14271361.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有
迷上了代码!