给定一个单词数组 words
和一个长度 maxWidth
,重新排版单词,使其成为每行恰好有 maxWidth
个字符,且左右两端对齐的文本。
你应该使用 “贪心算法” 来放置给定的单词;也就是说,尽可能多地往每行中放置单词。必要时可用空格 ' '
填充,使得每行恰好有 maxWidth 个字符。
要求尽可能均匀分配单词间的空格数量。如果某一行单词间的空格不能均匀分配,则左侧放置的空格数要多于右侧的空格数。
文本的最后一行应为左对齐,且单词之间不插入额外的空格。
注意:
- 单词是指由非空格字符组成的字符序列。
- 每个单词的长度大于 0,小于等于 maxWidth。
- 输入单词数组
words
至少包含一个单词。
示例 1:
输入: words = ["This", "is", "an", "example", "of", "text", "justification."], maxWidth = 16 输出: [ "This is an", "example of text", "justification. " ]
示例 2:
输入:words = ["What","must","be","acknowledgment","shall","be"], maxWidth = 16 输出: [ "What must be", "acknowledgment ", "shall be " ] 解释: 注意最后一行的格式应为 "shall be " 而不是 "shall be", 因为最后一行应为左对齐,而不是左右两端对齐。 第二行同样为左对齐,这是因为这行只包含一个单词。
示例 3:
输入:words = ["Science","is","what","we","understand","well","enough","to","explain","to","a","computer.","Art","is","everything","else","we","do"],maxWidth = 20 输出: [ "Science is what we", "understand well", "enough to explain to", "a computer. Art is", "everything else we", "do " ]
提示:
1 <= words.length <= 300
1 <= words[i].length <= 20
words[i]
由小写英文字母和符号组成1 <= maxWidth <= 100
words[i].length <= maxWidth
class Solution {
string blank(int n) { return string(n, ' '); } // 创建一个由n个空格组成的字符串
string join(vector<string>& words, int left, int right, string sep) { // 将单词数组中[left, right)的元素用sep字符串连接起来
string s = words[left];
for (int i = left + 1; i < right; i++) {
s += sep + words[i];
}
return s;
}
public:
vector<string> fullJustify(vector<string>& words, int maxWidth) {
vector<string> ans;
int right = 0, n = words.size();
while (true) {
int left = right;
int sumLen = 0;
while (right < n &&
sumLen + words[right].length() + right - left <= maxWidth) {
sumLen += words[right++].length();
}
if (right == n) { // 处理最后一行
string s = join(words, left, n, " "); // 将剩余的单词用空格连接起来
ans.emplace_back(s + blank(maxWidth - s.length())); // 将剩余空格添加到行的末尾
return ans;
}
int numWords = right - left;
int numSpaces = maxWidth - sumLen;
if (numWords == 1) { // 处理只有一个单词的情况
ans.emplace_back(words[left] + blank(numSpaces)); // 在单词后面添加剩余的空格
continue;
}
int avgSpaces = numSpaces / (numWords - 1); // 平均每个单词之间的空格数
int extraSpaces = numSpaces % (numWords - 1); // 余下的空格数
// 将额外的空格均匀分配给前extraSpaces个单词,剩下的平均分配给每个单词
string s1 = join(words, left, left + extraSpaces + 1,
blank(avgSpaces + 1)); // 拼接额外加一个空格的单词
string s2 = join(words, left + extraSpaces + 1, right,
blank(avgSpaces)); // 拼接其余单词
ans.emplace_back(s1 + blank(avgSpaces) + s2); // 将结果添加到答案数组中
}
}
};
首先创建了一个blank函数用于返回n个空格的字符串,然后创建了一个join函数,目的是用sep连接单词。
之后我创造了一个string类型的向量接收答案,right表示当前行的最右边的单词下标,left则表示最左边的单词下标,n存储总共的单词数,接下来进入while循环,初始化left=right,sumLen用于记录该行一共使用了几个字母,先初始化为0,当right<n既还没遍历到最后一个单词,并且目前占用的字母数加接下来要加入的单词字母数加上空格数还没超过该行的最大字符长度时,sumLen加入下一个单词的字符长度,right++,既指向在下一个单词的下标。这样就成功取得了该行的字母总数。
遍历完这一行之后,如果right==n既right已经指向了最后一个单词的下标,也就是现在的这一行就是最后一行了,用s存储剩下的这些单词,并用空格连接,然后将s以及剩下需要的空格加入ans的末尾,返回ans.
用numWords存储该行的单词数目,numSpaces存储该行一共需要补全的空格数目。
如果该行只有一个单词,那么将该单词加入ans最后,并加入剩余需要的空格,跳过本次循环。
用avgSpaces存储平均每两个单词之间的空格数,extraSpaces存储余下的空格数,由于题目要求,如果有余下的空格,则平均分给左边,所以现在创建两个字符串,第一个是左边的字符串,通过join函数用avgSpaces+1个空格将前面extraSpaces+1个单词连接,第二个是右边的字符串,同理用avgSpaces个空格将最后的单词连接,最后连接这两个字符串并加入ans的后面,这里注意,这两个字符串中间也要用空格连接。