问题描述
用python解决问题的方法有很多:函数法、暴力法以及一些常见的算法等等,而且python 的很多基础题用函数来解决是非常快捷的。熟练掌握函数的方法和技巧能够更好地解决问题,思路更加清晰。下面以leetcode”最长公共前缀”案例简单介绍zip函数的应用。
编写一个函数来查找字符串数组中的最长公共前缀。如果不存在公共前缀,返回空字符串 ""
示例:
输入: ["flower","flow","flight"]
输出: "fl"
解决方案
zip()函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的对象,最后需要转换成list()来输出列表。zip(*)可以将元组解压为列表。
代码示例:
a = [1,2,3] b = [4,5,6] print(list(zip(a,b))) a1,a2 = zip(*zip(a,b)) print(list(a1)) print(list(a2))
[(1, 4), (2, 5), (3, 6)] [1, 2, 3] [4, 5, 6] |
要解决最长的公共前缀,用zip函数是最简单的。可以把输入的字符串数组看成一个压缩包,用zip(*)对压缩包进行解压,再判断解压出来的数组列表是否为重复元素数组,如果是则返回前缀。
解题思路:
首先给a赋值一个空的字符串""若前缀不存在返回空字符串a;
用for循环对解压的字符串数组进行遍历 ;
set()函数对解压数组求不重复数组;
用if条件判断不重复数组的长度是否为1;
若为1,则将数组的元素加到a中;
最后返回a。
代码示例:
class Solution: def longestCommonPrefix(self, strs: List[str]) -> str: a = "" for i in zip(*strs): if len(set(i)) == 1: a = a + i[0] else: break return a |
结语
zip()是可迭代对象,使用时必须将其包含在一个list中,才能输出结果。熟练掌握函数知识,善于用函数快速求解问题。Python的函数有很多,一个一个的去记又很麻烦,所以需要通过实际问题来掌握函数并巧妙的应用。