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

最新推荐文章于 2024-09-21 16:35:48 发布

'楓

最新推荐文章于 2024-09-21 16:35:48 发布

阅读量62

点赞数

分类专栏： LeetCode 文章标签：算法

本文链接：https://blog.csdn.net/weixin_43812609/article/details/113274193

版权

LeetCode 专栏收录该内容

34 篇文章 0 订阅

订阅专栏

题目描述：
在这里插入图片描述
将二维数组逆时针旋转45°，可以将整个矩阵看作一个二叉搜索树，其中右上角和左下角可分别看作根节点。

左下角看作根节点：若当前值大于target则舍弃该行，若当前值小于target则舍弃该列。

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

右上角看作根节点：若target大于本节点则舍弃该行进入下一行，若小于target则舍弃该列进入前一列

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