[LeetCode]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”

方法

一次遍历

class Solution {
    public String truncateSentence(String s, int k) {
        int count = 1;
        int index = 0;
        for(int i =0;i<s.length();i++){
            if(s.charAt(i) == ' '){
                count++;
            }
            if(count == k){
                index = i;
            }
        }
        return s.substring(0,index+1);
    }
}

1. 使用count记录遍历到几个空格符，也代表遍历过几个单词
2. 用index记录最后位置的下标用于截取字符串

注意：

1. substring() 方法第二个参数是开区间，因此需要 index+1