[leetCode] 1816. Truncate Sentence (Python)

▼ 문제 바로가기 (링크) ▼ 

https://leetcode.com/problems/truncate-sentence/

Truncate Sentence - LeetCode

---

문자열 s 와 정수 k가 주어진다.

문자열을 공백 기준으로 나누었을 때 0번째 원소부터 k-1 번째까지를 반환한다.

이 때 자료형은 문자열이며, 원본과 같이 원소 사이의 공백을 포함한다.

class Solution:
    def truncateSentence(self, s: str, k: int) -> str:

        tmp = s.split()       
        answer = ""
        
        for i in range(k):
            answer += (tmp[i] + " ")
        answer = answer.rstrip()    
        
        return answer

먼저 임시 변수 tmp에 공백 기준으로 자른 문자열을 담았다.

for문으로 정답을 담은 answer 변수에 공백과 함께 추가하고

마지막 요소 끝의 공백을 제거한 후 반환했다.