题目要求
给你两个字符串 word1 和 word2 。请你从 word1 开始,通过交替添加字母来合并字符串。如果一个字符串比另一个字符串长,就将多出来的字母追加到合并后字符串的末尾。
返回 合并后的字符串
来源:力扣
思路
因为输入的两个字符串word1和word2长短会有可能不同,首先对输入的两个字符串长度进行对比,首先以长度小的开始遍历得到字符串a,然后再遍历长度长的字符串剩下的字符得到b,返回a+b得到输出。
class Solution:
def mergeAlternately(self, word1: str, word2: str) -> str:
if len(word1)>len(word2):
a=''
b=''
for i in range(0,len(word2)):
a=a+word1[i]+word2[i]
for k in range(len(word2),len(word1)):
b=b+word1[k]
return a+b
if len(word1)==len(word2):
a=''
b=''
for i in range(len(word1)):
a=a+word1[i]+word2[i]
return a
if len(word1)<len(word2):
a=''
b=''
for i in range(len(word1)):
a=a+word1[i]+word2[i]
for k in range(len(word1),len(word2)):
b=b+word2[k]
return a+b
提交之后发现用时比较长,参考其他大神代码了解到了一个新的函数---itertools模块的zip_longest函数。
zip_longest具体可以用来对列表的一 一对应,如果列表的长度不一致,则其会选择最长的那个列表,并将没有的填充为None(这个可以自己传参时设置)
class Solution:
def mergeAlternately(self, word1: str, word2: str) -> str:
return ''.join(a + b for a, b in zip_longest(word1, word2, fillvalue=''))
'''
word1='abcde'
word2='fgh'
zip_longest=(('a','f'),('b','g'),('c','h'),('d',' '),('e',' '))
最后输出afbgchde
'''
fillvalue=‘ ’作用是长的字符串后面的字符与空格一 一对应起来,最后输出空格不影响结果。