[LeetCode] 2109. Adding Spaces to a String - Python

 

난이도: Medium

 

문제 설명

---

You are given a 0-indexed string s and a 0-indexed integer array spaces that describes the indices in the original string where spaces will be added. Each space should be inserted before the character at the given index.

 

• For example, given s = "EnjoyYourCoffee" and spaces = [5, 9], we place spaces before 'Y' and 'C', which are at indices 5 and 9 respectively. Thus, we obtain "Enjoy Your Coffee".

Return the modified string after the spaces have been added.

 

문제 예시

---

Example 1

Input: s = "LeetcodeHelpsMeLearn", spaces = [8,13,15]
Output: "Leetcode Helps Me Learn"
Explanation: 
The indices 8, 13, and 15 correspond to the underlined characters in "LeetcodeHelpsMeLearn".
We then place spaces before those characters.

 

Example 2

Input: s = "icodeinpython", spaces = [1,5,7,9]
Output: "i code in py thon"
Explanation:
The indices 1, 5, 7, and 9 correspond to the underlined characters in "icodeinpython".
We then place spaces before those characters.

 

Example 3

Input: s = "spacing", spaces = [0,1,2,3,4,5,6]
Output: " s p a c i n g"
Explanation:
We are also able to place spaces before the first character of the string.

 

 

제한 사항

---

• 1 <= s.length <= 3 * 10^5
• s consists only of lowercase and uppercase English letters.
• 1 <= spaces.length <= 3 * 10^5
• 0 <= spaces[i] <= s.length - 1
• All the values of spaces are strictly increasing.

 

✏️ Solution(솔루션)

---

class Solution:
    def addSpaces(self, s: str, spaces: List[int]) -> str:
        s = list(s)

        for space in spaces:
            s[space] = f' {s[space]}'
        
        return ''.join(s)

 

먼저, s 문자열을 list로 바꿔주었다. (ex. 'abc' → ['a', 'b', 'c'])

 

그 후 바꾼 list에서 spaces에 저장된 위치의 요소에 앞에 공백을 추가하여 저장했다.

예를 들어, 만약 input이 s = 'abc'이고 spaces = [1] 이라고 한다면 ['a', ' b', 'c']와 같이 저장된다.

 

마지막으로 이 list를 join함수를 통해 문자열로 변환하여 반환하면 된다.

---

[LeetCode] 2109. Adding Spaces to a String