给定一个单词数组 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 {
public List<String> fullJustify(String[] words, int maxWidth) {
List<String> ans = new ArrayList<String>(); // 结果集合,存放最终的文本行
int right = 0, n = words.length; // right表示当前行的最后一个单词的下一个位置,n是单词数组的长度
while (true) {
int left = right; // 当前行的第一个单词在 words 数组中的索引位置
int sumLen = 0; // 统计当前行单词长度之和,不包括单词之间的空格
// 循环确定当前行可以放多少个单词,保证单词之间至少有一个空格
while (right < n && sumLen + words[right].length() + right - left <= maxWidth) {
sumLen += words[right].length();
right++;
}
// 当前行是最后一行:单词左对齐,单词之间只有一个空格,在行末填充剩余空格
if (right == n) {
StringBuffer sb = join(words, left, n, " "); // 将最后一行所有单词用一个空格连接
sb.append(blank(maxWidth - sb.length())); // 行末填充剩余的空格
ans.add(sb.toString()); // 将填好的行添加到结果集中
return ans; // 返回最终结果
}
int numWords = right - left; // 当前行单词个数
int numSpaces = maxWidth - sumLen; // 当前行需要填充的总空格数
// 当前行只有一个单词:该单词左对齐,在行末填充剩余空格
if (numWords == 1) {
StringBuffer sb = new StringBuffer(words[left]); // 创建一个 StringBuffer,用于存放当前行的文本
sb.append(blank(numSpaces)); // 在单词后面填充剩余的空格
ans.add(sb.toString()); // 将填好的行添加到结果集中
continue; // 继续处理下一行
}
// 当前行有多个单词
int avgSpaces = numSpaces / (numWords - 1); // 平均每个单词之间的空格数
int extraSpaces = numSpaces % (numWords - 1); // 需要额外添加一个空格的单词数量
StringBuffer sb = new StringBuffer();
sb.append(join(words, left, left + extraSpaces + 1, blank(avgSpaces + 1))); // 将额外加一个空格的单词拼接到当前行中
sb.append(blank(avgSpaces)); // 填充剩余的空格到当前行
sb.append(join(words, left + extraSpaces + 1, right, blank(avgSpaces))); // 拼接剩余的单词到当前行
ans.add(sb.toString()); // 将填好的行添加到结果集中
}
}
// 返回长度为 n 的由空格组成的字符串
public String blank(int n) {
StringBuffer sb = new StringBuffer();
for (int i = 0; i < n; ++i) {
sb.append(' '); // 将空格字符添加到 StringBuffer 中
}
return sb.toString(); // 返回由空格组成的字符串
}
// 返回用 sep 拼接 [left, right) 范围内的 words 组成的字符串
public StringBuffer join(String[] words, int left, int right, String sep) {
StringBuffer sb = new StringBuffer(words[left]); // 以第一个单词开始创建 StringBuffer 对象
for (int i = left + 1; i < right; ++i) {
sb.append(sep); // 在单词之间添加 sep 分隔符
sb.append(words[i]); // 将单词添加到 StringBuffer 中
}
return sb; // 返回拼接好的 StringBuffer 对象
}
}
解题思路:
-
初始化及循环:
right
和left
分别表示当前行的最后一个单词的下一个位置和当前行的第一个单词的位置。sumLen
用于统计当前行单词长度总和,不包括单词之间的空格。
-
确定当前行单词数:
- 在保证单词之间至少有一个空格的前提下,通过增加
right
的方式确定当前行可以容纳的单词数。
- 在保证单词之间至少有一个空格的前提下,通过增加
-
处理最后一行:
- 如果
right
等于单词总数n
,表示当前处理到了最后一行,最后一行的处理方式是将剩余的单词左对齐,并在行末填充剩余的空格,然后将其添加到结果集中并返回。
- 如果
-
处理单个单词行:
- 如果当前行只有一个单词,直接将该单词左对齐,并在其后填充剩余的空格,然后添加到结果集中。
-
处理多个单词行:
- 计算出当前行需要的总空格数
numSpaces
和单词数numWords
。 - 计算平均每个单词之间的空格数
avgSpaces
和需要额外一个空格的单词数量extraSpaces
。 - 使用
join
函数将单词和空格拼接成完整的当前行文本,并将其添加到结果集中。
- 计算出当前行需要的总空格数
-
辅助函数:
blank(int n)
:返回一个长度为n
的由空格组成的字符串。join(String[] words, int left, int right, String sep)
:将words
数组中[left, right)
范围内的单词用sep
连接成一个字符串。
这种算法通过逐行处理文本,根据不同情况(是否为最后一行、是否只有一个单词)应用不同的对齐方式,最终生成完整且格式正确的文本行。