二维数组的查找

二维数组中的查找_牛客网

https://www.nowcoder.com/practice/abc3fe2ce8e146608e868a70efebf62e?tpId=13&tqId=11154&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

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

分析:从左下角元素往上查找,右边元素是比这个元素大,上边是的元素比这个元素小。于是,target比这个元素小就往上找,比这个元素大就往右找。如果出了边界,则说明二维数组中不存在target元素。
AC代码:

def Find(self, target, array):
    # write code here
    row = len(array) - 1
    col = len(array[0]) - 1
    i, j = row, 0
    while i >= 0 and j <= col:
        if target < array[i][j]:
            i -= 1
        elif target > array[i][j]:
            j += 1
        else:
            return True
    return False

错误使用二分法,误以为第二行第二列的值一定比第三行第一列大:

    row = len(array) - 1
    col = len(array[0]) - 1
    i = int((row + 0) / 2)
    j = int((col + 0) / 2)
    while i >=0 and j >= 0 and i <= row and j <= col:
        if target < array[i][j]:
            i = int((i + 0) / 2)
            j = int((j + 0) / 2)
        elif target > array[i][j]:
            i = int((row + i) / 2)
            j = int((col + j) / 2)
        else:
            return True
    return False

二分法正确思路:
把每一行看成有序递增的数组,利用二分查找,通过遍历每一行得到答案,时间复杂度是nlogn。

    row = len(array) - 1
    col = len(array[0]) - 1
    i = 0
    while i <= row:
        low = 0
        high = col
        while low <= high:
            mid = int((low + high) / 2)
            if target < array[i][mid]:
                high = mid - 1
            elif target > array[i][mid]:
                low = mid + 1
            else:
                return True
        i += 1
    return False
全部评论

相关推荐

03-21 10:53
复旦大学 Java
大家好,我是@程序员花海,眼下&nbsp;26&nbsp;届春招、27&nbsp;届暑期实习全面开启,后端卷到没边,AI&nbsp;Agent的岗位占主导,很多牛友在我的评论区留言,想让我出一份Agent学习路线。我特意去看了下,打开淘天的招聘页面,以校招为例,一眼望去全是AI相关的岗位,只能说之后绝大多数岗位都会快速推进AI的落地和实践。之前写过&nbsp;Java&nbsp;后端&nbsp;3&nbsp;个月抢救路线https://www.nowcoder.com/discuss/824693499982315520?sourceSSR=users,也收到了牛友们的强烈好评,这次专门给后端转&nbsp;Agent做一套最少必要知识路线——&nbsp;不堆概念、不啃论文,只学面试必问、项目...
在职牛马didi:这篇路线整理得很系统,把后端知识映射到Agent体系这个思路特别实用。我自己也是从Java转做AI的,感触很深:工程底子扎实的人转Agent确实有优势,RAG和工具编排这些核心能力本质上都是后端逻辑的延伸。我们团队在做天猫的AI应用落地,方向跟你这篇路线里的企业级RAG和Agent系统很接近。暑期实习还在招AI应用研发工程师,JD可以参考看看跟你背景是否匹配:https://www.nowcoder.com/jobs/detail/440929?jobId=440929
软件开发投递记录
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务