0068：文本左右对齐（★★）

力扣第 68 题

题目

给定一个单词数组 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

分析

模拟：

• 遍历单词放到当前行，放不下了就调整空格分配输出当前行，从下一行重新开始
• 分配空格：
  • 先得到空格总数和单词数
  • 求得平均空格数 q 和余数 r
  • 前 r 个间隔分 q+1 个空格，后面的分 q 个空格即可
• 注意一行只有一个单词和最后一行的特殊情况，补空格即可

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

class Solution:
    def fullJustify(self, words: List[str], maxWidth: int) -> List[str]:
        res,row,s = [],[],0
        for w in words:
            if s+len(row)+len(w)<=maxWidth:
                row.append(w)
                s += len(w)
            else:
                if len(row)==1:
                    res.append(row[0].ljust(maxWidth))
                else:
                    q,r = divmod(maxWidth-s,len(row)-1)
                    for i in range(len(row)-1):
                        row[i] += ' '*(q+(i<r))
                    res.append(''.join(row))
                row,s = [w],len(w)
        res.append(' '.join(row).ljust(maxWidth))
        return res

24 ms