1816. 截断句子

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

例如,“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"
class Solution {
public:
    string truncateSentence(string s, int k) {
        int end = 0, count = 0;
        int n = s.size() - 1;
        for(int i = 0; i <= s.size(); ++i){
            if(s[i] == ' '){
                ++count;
                if(count == k){
                    end = i;
                    break;
                }
            }
        }
        if(count < k){
            return s;
        }
        return s.substr(0, end);
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43599304/article/details/121743000