二维数组的查找

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

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
全部评论

相关推荐

2024-12-29 15:37
已编辑
西华大学 图像识别
程序员牛肉:去不了,大厂算法卡学历吧
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务