返回不同長度的兩個字符串中的交替字母

Question

我目前有一個代碼，它遞歸地取兩個字符串中的字母，並用交替字母返回新字。我想優化此代碼，以便如果第一個或第二個單詞更長，它仍然會返回較長字符串中的其餘字母。

def alt(s,t): 
    if len(s) != len(t): 
     return 
    elif s == '' and t == '': 
     return '' 
    else: 
     return s[0] + t[0] + alt(s[1:], t[1:]) 

所需的輸出：

>>> alt('hello','bye') 
'hbeylelo' 

Answer 1

只是測試s和t是空的，返回其他價值，如果他們中的一個是：

def alt(s, t): 
    if not s: 
     return t 
    elif not t: 
     return s 
    else: 
     return s[0] + t[0] + alt(s[1:], t[1:]) 

即使兩個s和t爲空，返回空字符串，這是一個完全有效的結束狀態。

您可以縮短這：

def alt(s, t): 
    if not (s and t): 
     return s + t 
    return s[0] + t[0] + alt(s[1:], t[1:]) 

所以在達到最終狀態時無論是s或t是空的（或者它們兩者都是）。

這將產生所需輸出：

>>> alt('hello', 'bye') 
'hbeylelo' 

迭代版本是：

from itertools import chain 
try: 
    # Python 2 
    from itertools import izip_longest as zip_longest 
except ImportError: 
    # Python 3 
    from itertools import zip_longest 

def alt_iterative(s, t): 
    return ''.join(chain.from_iterable(zip_longest(s, t, fillvalue=''))) 

它使用itertools.zip_longest() function做的大部分工作。

Answer 2

問題本身不是固有的遞歸。在這種情況下，迭代解決方案可能更簡單。

的Python 2：

from itertools import izip_longest 
def alt(s, t): 
    return ''.join(a + b for a, b in izip_longest(s, t, fillvalue='')) 

的Python 3：

from itertools import zip_longest 
def alt(s, t): 
    return ''.join(a + b for a, b in zip_longest(s, t, fillvalue='')) 

Answer 3

如果你喜歡使用itertools.chain和zip（或itertools.zip_longest/izip_longest）一個班輪：

# Python 2 
return ''.join(chain(*zip(s, t))) + s[len(t):] + t[len(s):] 
# or 
return ''.join(chain(*izip_longest(s, t, fillvalue=''))) 

# Python 3 
return ''.join(chain(*zip_longest(s, t, fillvalue='')))