1816.截断句子:
句子 是一个单词列表,列表中的单词之间用单个空格隔开,且不存在前导或尾随空格。每个单词仅由大小写英文字母组成(不含标点符号)。
例如,"Hello World"、"HELLO" 和 "hello world hello world" 都是句子。
给你一个句子 s 和一个整数 k ,请你将 s 截断 ,使截断后的句子仅含 前 k 个单词。返回 截断 s 后得到的句子。
class Solution:
def truncateSentence(self, s: str, k: int) -> str:
return " ".join(s.split(" ")[:k])
使用暴力方法,将字符串转化为列表,然后按照位置添加空格,连接成为字符串,中间会有部分用例无法通过,问题有很多。对python掌握熟悉的情况下一行代码即可解决。
官方解答:
由题意可知,除了最后一个单词,每个单词后面都跟随一个空格。因此我们可以通过统计空格与句子结尾的数目来统计单词数 count。当 count=k 时,将当前的下标记录到 end,返回句子 s 在 end 处截断的句子。
def truncateSentence(s, k):
#n为字符串长度,end,count记录单词数
n, end, count = len(s), 0, 0
#遍历字符串
for i in range(1, n + 1):
#如果当前索引等于字符串长度或者当前索引对应的字符为空格的话,往下执行
if i == n or s[i] == ' ':
#单词数+1
count += 1
#如果单词数达到要求返回的单词数
if count == k:
#i为单词数达到要求返回的单词数时,此时的字符串索引位置
end = i
break
#返回end索引位置以前的所有字符,不包括end索引处,即为要求的返回值
return s[:end]
print(truncateSentence("Hello how are you Contestant",4))
复杂度分析:
时间复杂度:O(N),其中 N 为句子s 的长度。遍历整个字符串需要O(N)。
空间复杂度:O(1)。注意返回值不计入空间复杂度。
想法:这道题目的聚焦点不应该在单词中,而是聚焦在空格中,以空格为主分析求解,当空格数达到要求个数时,该空格前边的单词数也会达到要求个数,此时我们标记空格的索引位置给end,从原字符串中返回,从头到end索引处即可。