LeetCode原题:最长公共前缀
编写一个函数来查找字符串数组中的最长公共前缀。如果不存在公共前缀,返回空字符串 “”。
示例 1:
输入: [“flower”,“flow”,“flight”]
输出: "fl
我的代码
def longestCommonPrefix(strs):
if len(strs) == 0:
return ''
if len(strs) == 1:
return strs[0]
maxlen = min(len(item) for item in strs)
commstr = []
for i in range(maxlen):
locStr = [item[i] for item in strs]
if len(set(locStr)) == 1:
commstr.append(locStr[0])
else:
break
if len(commstr) == 0:
return ''
else:
return ''.join(commstr)
这里先记录最短的字符串长度,然后用纵向查找的方法在每个字符串中,从第一位开始查找,如果相同则找下一个,直到找到最短的字符串长度的位置停止。
使用zip
def longestCommonPrefix(strs):
if len(strs) == 0:
return ''
if len(strs) == 1:
return strs[0]
commstr = []
for item in zip(*strs):
if len(set(item)) == 1:
commstr.append(item[0])
else:
break
if len(commstr) == 0:
return ''
else:
return ''.join(commstr)
这里使用zip(*strs)
,直接给出了纵向对,且元素个数与最短的列表一致,这样也不用记录最短字符串长度, 同时节省了纵向位置的遍历代码,更为简洁。
关于zip
>>> strs
['flower', 'flow', 'flight']
>>> for item in zip(strs):
... print(item)
...
('flower',)
('flow',)
('flight',)
>>> for item in zip(*strs):
... print(item)
...
('f', 'f', 'f')
('l', 'l', 'l')
('o', 'o', 'i')
('w', 'w', 'g')