不难想但是很难写,需要考虑情况。
我是先写出代码框架(先看一行可以加入接下来几个字母和对应空格,然后用空格数和字母数相除取模计算出字母间可以塞多少个空格,循环塞入这些空格和字母,添加进结果中),最后添加判断是否是最后一个字母的情况分支。
class Solution {
public:
vector<string> fullJustify(vector<string>& words, int maxWidth) {
int n=words.size();
vector<string> result;
int i=0;
while(i<n){
string s;
int num=0;
int sum=words[i].size();
int j=i+1;
for(j;j<n;j++){
if(sum+words[j].size()+num+1>maxWidth) break;
else{
num++;sum+=words[j].size();
}
}
if(j==n){
s.append(words[i]);
for(int w=i+1;w<=num+i;w++){
s.push_back(' ');
s.append(words[w]);
}
s.append(maxWidth-s.size(),' ');
result.push_back(s);
return result;
}
int xnum=maxWidth-sum;
int x=(num!=0)?(xnum/num):xnum;
int y=(num!=0)?(xnum%num):xnum;
s.append(words[i]);
if(num==0) s.append(maxWidth-words[i].size(),' ');
for(int w=i+1;w<=num+i;w++){
if(y>0) {s.append(x+1,' ');y--;}
else s.append(x,' ');
s.append(words[w]);
}
i+=num+1;
result.push_back(s);
}
return result;
}
};