给你两个字符串 word1
和 word2
。请你从 word1
开始,通过交替添加字母来合并字符串。如果一个字符串比另一个字符串长,就将多出来的字母追加到合并后字符串的末尾。
示例 1:
输入:word1 = "abc", word2 = "pqr" 输出:"apbqcr" 解释:字符串合并情况如下所示: word1: a b c word2: p q r 合并后: a p b q c r
示例 2:
输入:word1 = "ab", word2 = "pqrs" 输出:"apbqrs" 解释:注意,word2 比 word1 长,"rs" 需要追加到合并后字符串的末尾。 word1: a b word2: p q r s 合并后: a p b q r s
示例 3:
输入:word1 = "abcd", word2 = "pq" 输出:"apbqcd" 解释:注意,word1 比 word2 长,"cd" 需要追加到合并后字符串的末尾。 word1: a b c d word2: p q 合并后: a p b q c d
本题思路很明确,可以设置一个变量i来判断两个字符剩余的个数,i从0开始计数,当i小于两个字符串中的任意一个时,程序继续执行;否则,两段字符皆以遍历完毕。
class Solution:
def mergeAlternately(self, word1: str, word2: str) -> str:
n, m, i = len(word1), len(word2), 0
ans = ""
while i < n or i < m:
if i < n:
ans += word1[i]
if i < m:
ans += word2[i]
i += 1
return ans