Question

1768. 交替合并字符串

English Version

题目描述

给你两个字符串 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

 

提示：

• 1 <= word1.length, word2.length <= 100
• word1 和 word2 由小写英文字母组成

解法

方法一：直接模拟

我们遍历 word1, word2 两个字符串，依次取出字符，拼接到结果字符串中。Python 代码可以简化为一行。

时间复杂度 $O(m + n)$，其中 $m$ 和 $n$ 分别是两个字符串的长度。忽略答案的空间消耗，空间复杂度 $O(1)$。

class Solution:
  def mergeAlternately(self, word1: str, word2: str) -> str:
    return ''.join(a + b for a, b in zip_longest(word1, word2, fillvalue=''))

class Solution {
  public String mergeAlternately(String word1, String word2) {
    int m = word1.length(), n = word2.length();
    StringBuilder ans = new StringBuilder();
    for (int i = 0; i < m || i < n; ++i) {
      if (i < m) {
        ans.append(word1.charAt(i));
      }
      if (i < n) {
        ans.append(word2.charAt(i));
      }
    }
    return ans.toString();
  }
}

class Solution {
public:
  string mergeAlternately(string word1, string word2) {
    int m = word1.size(), n = word2.size();
    string ans;
    for (int i = 0; i < m || i < n; ++i) {
      if (i < m) ans += word1[i];
      if (i < n) ans += word2[i];
    }
    return ans;
  }
};

func mergeAlternately(word1 string, word2 string) string {
  m, n := len(word1), len(word2)
  ans := make([]byte, 0, m+n)
  for i := 0; i < m || i < n; i++ {
    if i < m {
      ans = append(ans, word1[i])
    }
    if i < n {
      ans = append(ans, word2[i])
    }
  }
  return string(ans)
}

function mergeAlternately(word1: string, word2: string): string {
  const ans: string[] = [];
  const [m, n] = [word1.length, word2.length];
  for (let i = 0; i < m || i < n; ++i) {
    if (i < m) {
      ans.push(word1[i]);
    }
    if (i < n) {
      ans.push(word2[i]);
    }
  }
  return ans.join('');
}

impl Solution {
  pub fn merge_alternately(word1: String, word2: String) -> String {
    let s1 = word1.as_bytes();
    let s2 = word2.as_bytes();
    let n = s1.len().max(s2.len());
    let mut res = vec![];
    for i in 0..n {
      if s1.get(i).is_some() {
        res.push(s1[i]);
      }
      if s2.get(i).is_some() {
        res.push(s2[i]);
      }
    }
    String::from_utf8(res).unwrap()
  }
}

char* mergeAlternately(char* word1, char* word2) {
  int m = strlen(word1);
  int n = strlen(word2);
  char* ans = malloc(sizeof(char) * (n + m + 1));
  int i = 0;
  int j = 0;
  while (i + j != m + n) {
    if (i < m) {
      ans[i + j] = word1[i];
      i++;
    }
    if (j < n) {
      ans[i + j] = word2[j];
      j++;
    }
  }
  ans[n + m] = '\0';
  return ans;
}