截断句子

104 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 1 天,点击查看活动详情

5722. 截断句子

题目:

句子 是一个单词列表,列表中的单词之间用单个空格隔开,且不存在前导或尾随空格。每个单词仅由大小写英文字母组成(不含标点符号)。

例如,"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"

分析

这种字符串截断的问题,就是python的最爱!

解题:

class Solution:
    def truncateSentence(self, s, k):
        list_s = s.split()
        return ' '.join(list_s[:min(k, len(list_s))])

思路与算法

由题意可知,除了最后一个单词,每个单词后面都跟随一个空格。因此我们可以通过统计空格与句子结尾的数目来统计单词数 countcount。当 count=kcount=k 时,将当前的下标记录到 endend,返回句子 ss 在 endend 处截断的句子。

复杂度分析

时间复杂度:O(N)O(N),其中 NN 为句子 ss 的长度。遍历整个字符串需要 O(N)O(N)。

空间复杂度:O(1)O(1)。注意返回值不计入空间复杂度。

此题比较简单,但是是我做出来的第一道题,我感觉很难。