截取字符串前K个单词

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”

来源：力扣（LeetCode）
链接：
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2、关键字

字符串，按空格分开截断，取前k个

3、思路

遍历整个单词字符串，使用一个临时变量tem把单词存起来，
使用一个vector把各单词存起来，
然后遍历这个vector数组，把结果搞成字符串返回。

4、notes

1、经常遍历然后统计出现的一个问题就是循环内的统计，统计不到最后一个，需要在跳出循环的时候再统计。
2、

# s.split(" ") 返回被" "分割的 字符串列表
# s.split(" ")[:k] 取前k个
"""
join() 函数使用
str = "-";
seq = ("a", "b", "c"); # 字符串序列
print str.join( seq );
a-b-c
"""

5、复杂度

时间：O（N)
空间：O(N)

6、code

class Solution {
public:
    string truncateSentence(string s, int k) {
        vector<string> vec;
        string tem;
        for(auto t : s){
            if(t == ' '){
                if(tem != ""){
                    vec.push_back(tem);
                    tem ="";
                }
            }
            else{
                tem += t;
            }
        }
        vec.push_back(tem); // 这里最后一下，也加进来，因为循环里没法处理。

        string res = "";
        if(vec.size() >=1 )  // 这里答案不严谨，不写这一句也能通过
            res = vec[0];
        for(int i = 1; i < k && i < vec.size(); i++){
            res += " " + vec[i];
        }
        return res;

    }
};

python

class Solution:
    def truncateSentence(self, s: str, k: int) -> str:
        return " ".join(s.split(" ")[:k])

# s.split(" ") 返回被" "分割的 字符串列表
# s.split(" ")[:k] 取前k个
"""
join() 函数使用
str = "-";
seq = ("a", "b", "c"); # 字符串序列
print str.join( seq );
a-b-c
"""