1816. 截断句子
难度 简单
句子 是一个单词列表,列表中的单词之间用单个空格隔开,且不存在前导或尾随空格。每个单词仅由大小写英文字母组成(不含标点符号)。
例如,"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 中的单词之间由单个空格隔开
- 不存在前导或尾随空格
题解
class Solution { public String truncateSentence(String s, int k) { String [] strArray = s.split(" "); String ans=""; int first=1; for(int i=0;i<k;i++){ if(first==1){ ans+=strArray[i]; first=0; }else{ ans+=" "+strArray[i]; } } return ans; } }
官方
方法一:遍历
思路与算法
由题意可知,除了最后一个单词,每个单词后面都跟随一个空格。因此我们可以通过统计空格与句子结尾的数目来统计单词数 count。当 count=k 时,将当前的下标记录到end,返回句子s 在end 处截断的句子。
代码
class Solution { public String truncateSentence(String s, int k) { int n = s.length(); int end = 0, count = 0; for (int i = 1; i <= n; i++) { if (i == n || s.charAt(i) == ' ') { count++; if (count == k) { end = i; break; } } } return s.substring(0, end); } }
复杂度分析
- 时间复杂度:O(N),其中 NN 为句子 \textit{s}s 的长度。遍历整个字符串需要 O(N)。
- 空间复杂度:O(1)。注意返回值不计入空间复杂度。