题解 | #矩阵元素查找#
矩阵元素查找
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]