【力扣·每日一题】1816. 截断句子(模拟)

linkkk

题意

思路

遍历整个字符串$s$，判断当前字符；
如果当前字符为空格，记录单词的个数并且保存当前单词；
否则，将字母添加到当前单词里并且继续遍历。
如果单词的个数大于$k$就跳出循环。

代码

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