一、题设
给定一个单词数组 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.几个一组?如何判断?:遍历数组,当累计长度大于规定的maxWidth则为一组,但在里面要加上默认的单词之间的一个字符。(如有三个单词那么加2再判断)
2.分组后的空格个数有几种情况?
情况一:当最后一行时:全部单词左对齐,中间空一格即可。
情况二:当非最后一行但只有一个单词时:该单词左对齐,后全补空格
情况三:当非最后一行且不只有一个单词时:首先要计算单词之间平均的空格个数avg,若除不尽则还需要计算剩余的空格个数extra,然后将前extra个单词之间的空格数设为(1+avg),将后面的单词之间的空格数设为avg,最后补上中间的一个avg个数的字符串即可。
三、代码实现
def fullJustify(self,words, maxWidth):
def blank(n):
return " " * n
right , n = 0 , len(words)
res = []
while True:
lineNum = 0
left = right # left代表每一行的开头在words中的下标
while right < n and lineNum + len(words[right]) + right - left <= maxWidth:
lineNum += len(words[right])
right += 1
# 如果到了最后一行,全部左对齐然后补空格
if right == n:
tmp = " ".join(words[left:])
res.append(tmp + blank(maxWidth - len(tmp)))
break
# 计算总间隔和单词个数
wordsnum = right - left
spannum = maxWidth - lineNum
# 若非最后一行只有一个数字
if wordsnum == 1:
tmp = words[left]
res.append(tmp + blank(maxWidth - len(words[left])))
# 若非最后一行不止一个数字
else:
avgspan = spannum // (wordsnum - 1)
extraspan = spannum % (wordsnum - 1)
# 如果某一行单词间的空格不能均匀分配,则左侧放置的空格数要多于右侧的空格数。
before = blank(avgspan + 1).join(words[left : left + extraspan + 1])
after = blank(avgspan).join(words[left+extraspan+1:right])
res.append(before + blank(avgspan)+after)
return res