句子 是一个单词列表,列表中的单词之间用单个空格隔开,且不存在前导或尾随空格。每个单词仅由大小写英文字母组成(不含标点符号)。
- 例如,
"Hello World"
、"HELLO"
和"hello world hello world"
都是句子。
给你一个句子 s
和一个整数 k
,请你将 s
截断 ,使截断后的句子仅含 前 k
个单词。返回 截断 s
后得到的句子。
示例 1:
输入:s = "Hello how are you Contestant", k = 4 输出:"Hello how are you" 解释: s 中的单词为 ["Hello", "how" "are", "you", "Contestant"] 前 4 个单词为 ["Hello", "how", "are", "you"] 因此,应当返回 "Hello how are you"
示例 2:
输入:s = "What is the solution to this problem", k = 4 输出:"What is the solution" 解释: s 中的单词为 ["What", "is" "the", "solution", "to", "this", "problem"] 前 4 个单词为 ["What", "is", "the", "solution"] 因此,应当返回 "What is the solution"
示例 3:
输入:s = "chopper is not a tanuki", k = 5 输出:"chopper is not a tanuki"
示例代码1:
class Solution:
def truncateSentence(self, s: str, k: int) -> str:
lst = s.split()
return ' '.join(lst[:k])
示例代码2【遍历】:
class Solution:
def truncateSentence(self, s: str, k: int) -> str:
n, end, count = len(s), 0, 0
for i in range(1, n + 1):
if i == n or s[i] == ' ':
count += 1
if count == k:
end = i
break
return s[:end]
思路与算法:
由题意可知,除了最后一个单词,每个单词后面都跟随一个空格。因此可以通过统计空格与句子结尾的数目来统计单词数 count。当 count=k 时,将当前的下标记录到 end,返回句子 s 在 end 处截断的句子。
复杂度分析:
- 时间复杂度:O(N),其中 NN 为句子 s 的长度。遍历整个字符串需要 O(N)。
- 空间复杂度:O(1)。注意返回值不计入空间复杂度。