leetcode刷题python之文本左右对齐

最新推荐文章于 2024-03-22 08:30:00 发布

leileii

最新推荐文章于 2024-03-22 08:30:00 发布

阅读量369

点赞数

分类专栏： leetcode_python

本文链接：https://blog.csdn.net/leileii/article/details/102817731

版权

该博客探讨了如何使用Python解决LeetCode中的文本对齐问题，通过编写函数确定单词排列和空格插入策略。

摘要由CSDN通过智能技术生成

思路：
先写小函数来确定一行中需要放几个单词（通过返回单词列表的开头和结尾），再确定空格的添加方法。

class Solution:
    def fullJustify(self, words: List[str], maxWidth: int) -> List[str]:
        i = 0
        n = len(words)
        res = []
        def one_row(i):
            left = i
            cur_row_len = len(words[i])
            i += 1
            while i < n:
                if cur_row_len + len(words[i]) + 1 <= maxWidth:
                    cur_row_len += (len(words[i]) + 1)
                    i += 1
                else:
                    break
            return left, i
        while i < n:
            left, i = one_row(i)
            one_row_words = words[left: i]
            if i == n:
                res.append(" ".join(one_row_words) + " " * (maxWidth - len(" ".join(one_row_words))))
                break
            num_space = maxWidth - len(