算法练习笔记-zip使用

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')