2109-向字符串添加空格

Raphael Liu Lv10

给你一个下标从 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 中的所有值 严格递增

方法一:双指针

思路与算法

我们可以使用两个指针分别遍历字符串 s 和数组 spaces。记前者的指针为 i,后者的指针为 ptr:当 spaces}[\textit{ptr}] 恰好与 i 相等时,我们在答案字符串的末尾放入一个空格,并将 ptr 向右移动一个位置。

此外,我们还需要在答案字符串的末尾放入 s[i],并将 i 向右移动一个位置。在两个指针全部遍历完成后,我们就得到了修改后的字符串。

代码

[sol1-C++]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution {
public:
string addSpaces(string s, vector<int>& spaces) {
int n = s.size();
string ans;
ans.reserve(n + spaces.size());

int ptr = 0;
for (int i = 0; i < n; ++i) {
if (ptr < spaces.size() && spaces[ptr] == i) {
ans.push_back(' ');
++ptr;
}
ans.push_back(s[i]);
}
return ans;
}
};
[sol1-Python3]
1
2
3
4
5
6
7
8
9
10
11
12
class Solution:
def addSpaces(self, s: str, spaces: List[int]) -> str:
ans = list()

ptr = 0
for i, ch in enumerate(s):
if ptr < len(spaces) and spaces[ptr] == i:
ans.append(" ")
ptr += 1
ans.append(ch)

return "".join(ans)

复杂度分析

  • 时间复杂度:O(n + m),其中 n 是字符串 s 的长度,m 是数组 spaces 的长度。

  • 空间复杂度:O(1) 或 O(n + m),取决于使用的语言中字符串是否可以修改。

 Comments
On this page
2109-向字符串添加空格