原题在这里:矩阵中的路径
我个人一般喜欢在 LeetCode 上刷题,不过有时候《剑指 Offer 》的题那里没有,会去牛客网上做。 之前就发现这道题在牛客网上测试用例不全,以为会修复,没想到几个月过去了,还是这样。把问题发到讨论区也没人理,也找不到合适的反馈入口。
这道题是用 Python 写的,排行榜上 Python 的大部分答案,我都认为是错的。
在一个特殊的 TestCase 下:
matrix = 'ABEESFCSADME', rows=3, cols=4, path='SEE'
排行榜中大部分的答案都输出了错误的结果False
,正确应该是True
。
ABEE
SFCS
ADME
错误的答案都是这么写的,在 if 条件后直接 return 了,那么在四个方向判断中,如果先向下走,会导致提前 return False,而不会再向上延伸。
# -*- coding:utf-8 -*-
class Solution:
def hasPath(self, matrix, rows, cols, path):
# write code here
for i in range(rows):
for j in range(cols):
if matrix[i*cols+j] == path[0]:
if self.find(list(matrix),rows,cols,path[1:],i,j):
return True
return False
def find(self,matrix,rows,cols,path,i,j):
if not path:
return True
matrix[i*cols+j]='0'
if j+1<cols and matrix[i*cols+j+1]==path[0]:
return self.find(matrix,rows,cols,path[1:],i,j+1)
elif j-1>=0 and matrix[i*cols+j-1]==path[0]:
return self.find(matrix,rows,cols,path[1:],i,j-1)
elif i+1<rows and matrix[(i+1)*cols+j]==path[0]:
return self.find(matrix,rows,cols,path[1:],i+1,j)
elif i-1>=0 and matrix[(i-1)*cols+j]==path[0]:
return self.find(matrix,rows,cols,path[1:],i-1,j)
else:
return False
我觉得正确的解法应该是这样,应该保留四个方向的结果做或运算。
def hasPath(matrix, rows, cols, path):
# write code here
for i in range(rows):
for j in range(cols):
if matrix[i*cols + j] == path[0]:
if spread(list(matrix), rows, cols, path[1:], i, j):
return True
return False
def spread(matrix, rows, cols, path, i, j):
if not path:
return True
matrix[i*cols + j] = '-'
up, down, left, right = False, False, False, False
if j + 1 < cols and matrix[i * cols + j + 1] == path[0]:
down = spread(matrix, rows, cols, path[1:], i, j + 1)
if j - 1 >= 0 and matrix[i * cols + j - 1] == path[0]:
left = spread(matrix, rows, cols, path[1:], i, j - 1)
if i + 1 < rows and matrix[(i + 1) * cols + j] == path[0]:
right = spread(matrix, rows, cols, path[1:], i + 1, j)
if i - 1 >= 0 and matrix[(i - 1) * cols + j] == path[0]:
up = spread(matrix, rows, cols, path[1:], i - 1, j)
return up or down or left or right
上面两个代码均能通过牛客网的全部测试用例。而输出结果却不一致,我还是确信自己的写法是正确的。 有刷过此题的 V 友,或者算法大佬帮忙解答一下吗?就是想证明一下自己没有想错。
这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。
V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.