1816. 截断句子

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 中的单词之间由单个空格隔开
• 不存在前导或尾随空格

解题思路

s 中的单词之间由单个空格隔开，因此我们需要遍历s中的每一个字符，当遇到空格时直接跳过，当遇到字母时，将字母后面连续的没有被空格分隔开的字符全部遍历一遍，直到遇到空格时结束遍历，则说明我们已经找到了一个单词，如此类推，直到我们找到了题目要求的k个单词以后，跳出循环，并且记录下当前的下标，该下标之前的字符串就是我们需要截断的字符串，然后将该子串返回。

代码

class Solution {
public:
    string truncateSentence(string s, int k) {

        int i=0;

        while (k>0){
            if (isalpha(s[i]))
            {
                while (isalpha(s[i]))
                    i++;
                k--;
            }else i++;
        }
        return s.substr(0,i);
    }
};