题解 | #矩阵元素查找#

矩阵元素查找

http://www.nowcoder.com/practice/3afe6fabdb2c46ed98f06cfd9a20f2ce

由于矩阵中数的大小性质,位于右上角的数为该行的最大数,该列的最小数

将其与x比较,如果大于x,则表示该列后面的数都大于x,则j-=1

如果小于x,则表示该行的数都小于x,则i+=1

#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 
# @param mat int整型二维数组 
# @param n int整型 
# @param m int整型 
# @param x int整型 
# @return int整型一维数组
#
class Solution:
    def findElement(self , mat: List[List[int]], n: int, m: int, x: int) -> List[int]:
        # write code here
        i,j=0,m-1
        while i<n and j>=0:
            if mat[i][j]<x:
                i+=1
            elif mat[i][j]>x:
                j-=1
            else:
                return [i,j]
全部评论

相关推荐

评论
点赞
1
分享

创作者周榜

更多
牛客网
牛客企业服务