这个题说简单也很简单,主要还是思路,不然确实很难写~~
如果从做左上角或者右下角开始遍历,需要考虑两种情况,很麻烦。如果是从右上角或者左下角开始遍历,就比较简单,只需要考虑一种情况。
class Solution:
def findElement(self, mat, n, m, x):
res1 = 0
res2 = m-1
while res1 < n and res2 >= 0:
if mat[res1][res2] > x:
res2 -= 1
elif mat[res1][res2] < x:
res1 += 1
else:
return [res1, res2]
return []