leetcode_2109 向字符串添加空格

141 阅读1分钟

要求

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

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

例如,s = "EnjoyYourCoffee" 且 spaces = [5, 9] ,那么我们需要在 'Y' 和 'C' 之前添加空格,这两个字符分别位于下标 5 和下标 9 。因此,最终得到 "Enjoy Your Coffee" 。 请你添加空格,并返回修改后的字符串。

示例 1:

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

示例 2:

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

示例 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 中的所有值 严格递增

核心代码

class Solution:
    def addSpaces(self, s: str, spaces: List[int]) -> str:
        n = len(s)
        cnt = -1
        si = 0
        sn = len(spaces)
        res = []
        for i in range(n):
            cnt += 1
            if si < sn and cnt == spaces[si]:
                res.append(' ')
                si += 1
            res.append(s[i])
        return ''.join(res)

image.png

解题思路:这个题就是在指定的位置添加空格,循环遍历,注意这个地方cnt,原始字符串的下标位置,比较简单。