1816. 截断句子

leetcode力扣刷题打卡

题目：1816. 截断句子
描述：句子 是一个单词列表，列表中的单词之间用单个空格隔开，且不存在前导或尾随空格。每个单词仅由大小写英文字母组成（不含标点符号）。

例如，“Hello World”、“HELLO” 和 “hello world hello world” 都是句子。
给你一个句子 s​​​​​​ 和一个整数 k​​​​​​ ，请你将 s​​ 截断 ​，​​​使截断后的句子仅含 前 k​​​​​​ 个单词。返回 截断 s​​​​​​ 后得到的句子。

解题思路

1、根据空格来判断单词数；
2、注意处理尾部的空格。

原代码##

class Solution {
public:
    string truncateSentence(string s, int k) {
        string ans = "";
        for (int i = 0; i < s.size(); ++i) {
            if (s[i] == ' ') {
                k--;
            }
            if (k >= 1) {
                ans += s[i];
            }
            if (k == 0) {
                break;
            }

        }
        return ans;
    }
};