LeetCode每日一题 - 截断句子

题目：
句子 是一个单词列表，列表中的单词之间用单个空格隔开，且不存在前导或尾随空格。每个单词仅由大小写英文字母组成（不含标点符号）。
例如，“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”

C语言：

char * truncateSentence(char * s, int k)
{
    int i, flag=0;
    int lens = strlen(s);
    for(i=0;i<lens;i++)
    {
        if(s[i] == ' ')
            flag++;
        if(flag == k)
            break;
    }
    s[i] = '\0';
    return s;
}

执行用时0ms，内存消耗5.9MB。

Python3：

class Solution:
    def truncateSentence(self, s: str, k: int) -> str:
        lens=len(s)
        n=0
        cnt=0
        while n<lens:
            if s[n] == ' ':
                cnt+=1
            if cnt==k:
                break
            n+=1
        if cnt <= k-1:
            return s
        return s[:n]

执行用时40ms，内存消耗15MB。