LeetCode: 68. 文本左右对齐

549 阅读3分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第29天,点击查看活动详情

68. 文本左右对齐

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/te…

给定一个单词数组和一个长度 maxWidth,重新排版单词,使其成为每行恰好有 maxWidth 个字符,且左右两端对齐的文本。 你应该使用“贪心算法”来放置给定的单词;也就是说,尽可能多地往每行中放置单词。必要时可用空格 ' ' 填充,使得每行恰好有 maxWidth 个字符。 要求尽可能均匀分配单词间的空格数量。如果某一行单词间的空格不能均匀分配,则左侧放置的空格数要多于右侧的空格数。 文本的最后一行应为左对齐,且单词之间不插入额外的空格。

说明:

  • 单词是指由非空格字符组成的字符序列。
  • 每个单词的长度大于 0,小于等于 maxWidth。
  • 输入单词数组 words 至少包含一个单词。

示例1:

输入: words = ["This", "is", "an", "example", "of", "text", "justification."] maxWidth = 16

输出:

在这里插入图片描述

示例 2:

输入: words = ["What","must","be","acknowledgment","shall","be"] maxWidth = 16

输出:

在这里插入图片描述

解释: 注意最后一行的格式应为 "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

输出:

在这里插入图片描述

解法

贪心算法加双指针

  • 贪心算法是判断每一行最多有几个单词,毕竟是按照顺序添加
  • 双指针是来判断每行单词的个数

根据题干描述的贪心算法,对于每一行,我们首先确定最多可以放置多少单词,这样可以得到该行的空格个数,从而确定该行单词之间的空格个数。

根据题目中填充空格的细节,我们分以下三种情况讨论:

当前行是最后一行:单词左对齐,且单词之间应只有一个空格,在行末填充剩余空格; 当前行不是最后一行,且只有一个单词:该单词左对齐,在行末填充空格; 当前行不是最后一行,且不只一个单词:设当前行单词数为numWords,空格数为numSpaces 在这里插入图片描述

def blank(n: int) -> str:
    return ' ' * n


class Solution:
    def fullJustify(self, words: List[str], maxWidth: int) -> List[str]:
        ans = []
        right = 0
        n = len(words)
        while True:
            left = right  # 当前行的第一个单词在 words 的位置
            sumLen = 0
            while right < n and sumLen + len(words[right]) + right - left <= maxWidth:
                sumLen += len(words[right])
                right += 1
            
            # 这一行不能再进行添加单词时,开始分行判断,最后一行,不是最后一行
            # 1. 当前行是最后一行:单词左对齐,且单词之间应只有一个空格,在行末填充剩余空格
            if right == n:
                s = ' '.join(words[left: right])
                ans.append(s+blank(maxWidth-len(s)))
                break
            
            numWords = right - left
            numSpaces = maxWidth - sumLen

            # 2. 当前行不是最后一行,且该行只有一个单词时,在末尾填充剩余空格
            if numWords == 1:
                ans.append(words[left] + blank(numSpaces))
                continue
            
            # 3. 当前行不是最后一行,且该行不止一个单词时候,左侧单词放置的空格数要多于右侧的空格数
            avgSpaces = numSpaces // (numWords - 1)   # 平均空格数
            extraSpaces = numSpaces % (numWords - 1)  # 也可以说明需要多加一个空格的单词数
            s1 = blank(avgSpaces + 1).join(words[left: left + extraSpaces + 1])  # 拼接额外加一个空格的单词 左边的单词个数
            s2 = blank(avgSpaces).join(words[left + extraSpaces + 1: right])  # 拼接其余单词 右边的单词
            ans.append(s1 + blank(avgSpaces) + s2)
        return ans

复杂度分析

  • 时间复杂度:O(m)O(m),其中 m 是数组 words 中所有字符串的长度之和
  • 空间复杂度:O(m)O(m)