题意
思路
遍历整个字符串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; } };