一、概述
输入一个字符串数组和一个数字m,其中元素是一个个单词,对单词进行重新组合,将一定数量的单词按原来的顺序组合在一起,在其中插入若干空格,以保证:
不是最后一行的,保证字母数量加空格数量为m,单词间的空格数量尽可能相同,不相同的话,保证左边的较多,保证本行最后一个字符不是空格;
是最后一行的,保证,单词间的空格数量都为1,字母数量加空格小于m时,剩余空格全加到行尾。
有点复杂,总而言之就是对一句话进行排版,分为两端对齐的多行。
二、分析
首先要分成两种情况:
到最后一行和不到最后一行。
不到最后一行是普遍情况,先讨论这个。主要思路就是维护两个指针i和left,一个变量len。left指向当前行的第一个单词,i指向当前正在处理的单词。
对于正在处理的单词,有三种情况:
第一种,i指向的单词长度加上len仍然小于m,那么就在这个单词后插入一个空格,len加一加上单词长度,再判断下一个单词——为什么要再加一呢?我们知道,在一行中,单词之间至少插入一个空格,这里的加一就是把这个空格算上了。
第二种,i指向的单词长度加上len大于m,那么就到上一个单词为止,开始准备整理成行。这里有一个问题:最后一个单词后面不能有空格,别急,一会判断这个。
第三种,i指向的单词长度加上len等于m,那么同样到上一个单词为止。
现在后两种情况都是到上一个单词为止,但是第二种情况下,我们需要把上一个单词最后的空格删去,第三种情况下,可以把当前单词加上——注意后面没有空格。这样就确定了当前行中容纳的所有单词。然后就要插入空格了,插多少呢?个数是m-len。从第一个单词往后加到倒数第二个单词,要是还有空格,那就从头再继续。
当我们到了最后一行,需要做的就是直接在行末补空格直到长度为m即可。
代码如下:
class Solution {
public:
vector<string> fullJustify(vector<string>& words, int maxWidth) {
int len=0,left=0,right=0,i=0;
vector<string> res;
while(i<=words.size())
{
if(i<words.size()&&len+words[i].size()<maxWidth)
{
len+=(words[i].size()+1);
words[i]+=' ';
++i;
continue;
}
if(i==words.size()||len+words[i].size()>=maxWidth)
{
if(i<words.size()&&len+words[i].size()==maxWidth)
{
len+=words[i].size();
++i;
}
else
{
words[i-1].erase(words[i-1].size()-1);
--len;
}
int exs=maxWidth-len;
right=i-1;
string tmp;
int j=0;
if(i!=words.size())
{
for(j=left;exs>0;++j)
{
words[j]+=' ';
--exs;
if(j>=right-1||j>words.size()-1)
j=left-1;
}
}
else
{
while(exs>0)
{
words[words.size()-1]+=' ';
--exs;
}
}
for(int j=left;j<=right;++j)
tmp+=words[j];
res.push_back(tmp);
if(i==words.size())
break;
len=0;
left=i;
len=0;
}
}
return res;
}
};
总而言之抓住两点:
其一是单词之间必须至少有一个空格;其二是维护len来继续插入空格。
三、总结
题目本身不难,主要是边界条件。还有后面加空格的时机:是在挑选的时候加,还是在确定之后一起加?都可以。