题目描述
句子 是一个单词列表,列表中的单词之间用单个空格隔开,且不存在前导或尾随空格。每个单词仅由大小写英文字母组成(不含标点符号)。
例如,“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 <= s.length <= 500
- k 的取值范围是 [1, s 中单词的数目]
- s 仅由大小写英文字母和空格组成
- s 中的单词之间由单个空格隔开
- 不存在前导或尾随空格
题解思路
- 一次遍历
- 或者使用spilt分隔
题解代码
方法一#一次遍历
class Solution:
def truncateSentence(self, s: str, k: int) -> str:
n=len(s)
end=None
for i,c in enumerate(s):
if c==' ':
k-=1
if not k:
end=i
break
return s[0:end] if end is not None else s
方法二
class Solution:
def truncateSentence(self, s: str, k: int) -> str:
return ' '.joint(s.split(' ')[0:k])
str.split()
str.split(str=" "num=)
对指定字符串进行切割,函数可以传入两个参数 一个是切割的标识另一个则是切割的个数(默认为-1即 全部)如果num有指定值 则默认切割为num+1个
例子
txt = "Google#Runoob#Taobao#Facebook"
x = txt.split("#", 1)
print x
---------------------------------------------------------
['Google', 'Runoob#Taobao#Facebook']