Question

2109. 向字符串添加空格

English Version

题目描述

给你一个下标从 0 开始的字符串 s ，以及一个下标从 0 开始的整数数组 spaces 。

数组 spaces 描述原字符串中需要添加空格的下标。每个空格都应该插入到给定索引处的字符值 之前 。

• 例如，s = "EnjoyYourCoffee" 且 spaces = [5, 9] ，那么我们需要在 'Y' 和 'C' 之前添加空格，这两个字符分别位于下标 5 和下标 9 。因此，最终得到 "Enjoy _Y_our _C_offee" 。

请你添加空格，并返回修改后的字符串_。_

 

示例 1：

输入：s = "LeetcodeHelpsMeLearn", spaces = [8,13,15]
输出："Leetcode Helps Me Learn"
解释：
下标 8、13 和 15 对应 "Leetcode_H_elps_M_e_L_earn" 中加粗斜体字符。
接着在这些字符前添加空格。

示例 2：

输入：s = "icodeinpython", spaces = [1,5,7,9]
输出："i code in py thon"
解释：
下标 1、5、7 和 9 对应 "i_c_ode_i_n_p_y_t_hon" 中加粗斜体字符。
接着在这些字符前添加空格。

示例 3：

输入：s = "spacing", spaces = [0,1,2,3,4,5,6]
输出：" s p a c i n g"
解释：
字符串的第一个字符前可以添加空格。

 

提示：

• 1 <= s.length <= 3 * 105
• s 仅由大小写英文字母组成
• 1 <= spaces.length <= 3 * 105
• 0 <= spaces[i] <= s.length - 1
• spaces 中的所有值 严格递增

解法

方法一：双指针

我们可以用双指针 $i$ 和 $j$ 分别指向字符串 $s$ 和数组 $spaces$ 的头部，然后从头到尾遍历字符串 $s$，当 $i$ 等于 $spaces[j]$ 时，我们往结果字符串中添加一个空格，然后 $j$ 自增 1。接下来，我们将 $s[i]$ 添加到结果字符串中，然后 $i$ 自增 1。继续这个过程，直到遍历完字符串 $s$。

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

class Solution:
  def addSpaces(self, s: str, spaces: List[int]) -> str:
    ans = []
    j = 0
    for i, c in enumerate(s):
      if j < len(spaces) and i == spaces[j]:
        ans.append(' ')
        j += 1
      ans.append(c)
    return ''.join(ans)

class Solution {
  public String addSpaces(String s, int[] spaces) {
    StringBuilder ans = new StringBuilder();
    for (int i = 0, j = 0; i < s.length(); ++i) {
      if (j < spaces.length && i == spaces[j]) {
        ans.append(' ');
        ++j;
      }
      ans.append(s.charAt(i));
    }
    return ans.toString();
  }
}

class Solution {
public:
  string addSpaces(string s, vector<int>& spaces) {
    string ans = "";
    for (int i = 0, j = 0; i < s.size(); ++i) {
      if (j < spaces.size() && i == spaces[j]) {
        ans += ' ';
        ++j;
      }
      ans += s[i];
    }
    return ans;
  }
};

func addSpaces(s string, spaces []int) string {
  var ans []byte
  for i, j := 0, 0; i < len(s); i++ {
    if j < len(spaces) && i == spaces[j] {
      ans = append(ans, ' ')
      j++
    }
    ans = append(ans, s[i])
  }
  return string(ans)
}

function addSpaces(s: string, spaces: number[]): string {
  let ans = '';
  for (let i = 0, j = 0; i < s.length; i++) {
    if (j < spaces.length && i === spaces[j]) {
      ans += ' ';
      ++j;
    }
    ans += s[i];
  }
  return ans;
}

方法二

class Solution:
  def addSpaces(self, s: str, spaces: List[int]) -> str:
    ans = []
    i, j = len(s) - 1, len(spaces) - 1
    while i >= 0:
      ans.append(s[i])
      if j >= 0 and i == spaces[j]:
        ans.append(' ')
        j -= 1
      i -= 1
    return ''.join(ans[::-1])