【LeetCode】Day03

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索引处即可。