【劍指Offer】1.二維數組中的查找

工作一段時間後,發現數據結構和算法不像框架一樣對開發效率提高那麼明顯,但是這屬於內功層面,不像框架那麼變換那麼塊。因此,重新撿起《劍指offer》系列來鍛鍊編程的敏感性。

題目描述

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

在一個二維數組中(每個一維數組的長度相同),每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函數,輸入這樣的一個二維數組和一個整數,判斷數組中是否含有該整數。

解題思路

利用二維數組由上到下,由左到右遞增的規律,那麼選取右上角或者左下角的元素a[row][col]與target進行比較。

當target小於元素a[row][col]時,那麼target必定在元素a所在行的左邊,即col--;

當target大於元素a[row][col]時,那麼target必定在元素a所在列的下邊,即row++;

代碼

# -*- coding:utf-8 -*-
class Solution:
    # array 二維列表
    def Find(self, target, array):
        # write code here
        if array==[[]]:
            return False
        
        colNum = len(array[0])
        rawNum = len(array)
        
        if type(target)==float and type(array[0][0])==int:
            target = int(target)
        elif type(target) == int and type(array[0][0]) == float:
            target = float(target)
        elif type(target) != type(array[0][0]):
            return False
        
        rawIndex = 0
        colIndex = colNum-1
        
        while colIndex >=0 and rawIndex < rawNum:
            if target > array[rawIndex][colIndex]:
                rawIndex+=1
            elif target < array[rawIndex][colIndex]:
                colIndex-=1
            else:
                return True
        return False

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章