题目描述
给定一个单词数组和一个长度 maxWidth,重新排版单词,使其成为每行恰好有 maxWidth 个字符,且左右两端对齐的文本。
你应该使用“贪心算法”来放置给定的单词;也就是说,尽可能多地往每行中放置单词。必要时可用空格 ’ ’ 填充,使得每行恰好有 maxWidth 个字符。
要求尽可能均匀分配单词间的空格数量。如果某一行单词间的空格不能均匀分配,则左侧放置的空格数要多于右侧的空格数。
文本的最后一行应为左对齐,且单词之间不插入额外的空格。
说明:
单词是指由非空格字符组成的字符序列。
每个单词的长度大于 0,小于等于 maxWidth。
输入单词数组 words 至少包含一个单词。
示例:
输入:
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、如果选中的单词数为n,则中间间隙为n - 1,如果需要填充的空格数不能整除前面的间隙数量,余数为r,则往前面平均填充前r个间隙,每个间隙多加一个空格就好了。
假设max length = 16,则如下:
2、如果是单个单词,末尾填充空格,如果不是单个单词,中间对齐,两边不能有空格。
3、如果是最后一列,左对齐,单词中间空隙一律一个空格,不够的在末尾填充。
大体思路为:
代码实现
public static List<String> fullJustify(String[] words, int maxWidth) {
int[] arr = new int[words.length];
//将单词转换成数字
for (int i = 0; i < words.length; i++) {
arr[i] = words[i].length();
}
List<List<Integer>> list = new ArrayList<>();
int temp = 0;
Queue<Integer> queue = new ArrayDeque<>();
// 将单词数先分组,不超过maxWidth为一组
for (int i = 0; i < arr.length; i++) {
if (i == 0) {
temp = arr[i];
queue.add(i);
} else {
//因为如果不是末尾的情况每一个单词后面必然有一个空格
if (temp + 1 + arr[i] <= maxWidth) {
temp = temp + 1 + arr[i];
queue.add(i);
}else {
list.add(new ArrayList<>(queue));
queue.clear();
temp = arr[i];
queue.add(i);
}
}
}
if (!queue.isEmpty()) {
list.add(new ArrayList<>(queue));
}
//取出每一组数开始计算空格数
int length = list.size();
List<String> res = new ArrayList<>();
for (int i = 0; i < length; i++) {
List<Integer> p = list.get(i);
int pL = p.size();
int sum = 0;
//先计算出单词占的数量情况
for (int j = 0; j < pL; j++) {
sum += arr[p.get(j)];
}
int avg = 0,re = 0;
// 计算中间间隙数
if (pL > 1) {
avg = (maxWidth - sum) / (pL - 1);
re = (maxWidth - sum) % (pL - 1);
}else {
avg = (maxWidth - sum) / pL;
re = (maxWidth - sum) % pL;
}
StringBuffer sb = new StringBuffer();
//先计算最后一列的情况
if (i == length - 1) {
for (int j = 0; j < pL; j++) {
sb.append(words[p.get(j)]);
sb.append(" ");
}
int q = maxWidth - sum - pL;
for (int j = 0; j < q; j++) {
sb.append(" ");
}
if (sb.length() > maxWidth) {
sb.deleteCharAt(maxWidth);
}
res.add(sb.toString());
break;
}
//计算非最后一列情况
for (int j = 0,k = 0; j < pL; j++,k++) {
sb.append(words[p.get(j)]);
//填充空格数
//k是需要填充余数的情况,re是余数
if (k < re) {
for (int l = 0; l < avg + 1; l++) {
sb.append(" ");
}
}else {
/**
* 需要填充的情况
* 1.不是最后一列了或者只有一个单词的情况
*/
if (j != pL - 1 || pL == 1) {
for (int l = 0; l < avg; l++) {
sb.append(" ");
}
}
}
}
res.add(sb.toString());
}
return res;
}
性能分析
最差的情况是每行都只有一个单词,算上填充空格的时间,时间复杂度为O(nmaxlength),空间复杂度需要一个list保留结果,n个单词,每个单词maxlength,所以最大空间复杂度也是O(nmaxlength)。
有不足之处多多指教
公众号:“计算机基础爱好者”
关注公众号回复“leetcode”,有leet-java.pdf书籍