python-leetcode-806. 写字符串需要的行数

806. 写字符串需要的行数 - 力扣（LeetCode）

我们可以使用贪心算法来解决这个问题。具体步骤如下：

1. 维护两个变量：

   • lines：用于记录总共使用了多少行，初始化为 1。

   • width：用于记录当前行的已使用宽度，初始化为 0。

2. 遍历字符串 S 中的每个字符：

   • 获取该字符所需的宽度 w = widths[ord(c) - ord('a')]。

   • 判断 width + w 是否超过 100：

     • 若未超过，则累加 w 到 width。

     • 若超过，则 lines 增加 1，width 重新赋值为 w（新行开始）。

3. 最后，返回 [lines, width]，其中 lines 代表总行数，width 代表最后一行的宽度。

代码实现：

def numberOfLines(widths, S):
    lines = 1
    width = 0

    for c in S:
        w = widths[ord(c) - ord('a')]
        if width + w > 100:
            lines += 1
            width = w
        else:
            width += w

    return [lines, width]

复杂度分析：

• 遍历 S 一遍，时间复杂度为 O(N)（其中 N 为 S 的长度）。

• 仅使用了几个额外变量，空间复杂度为 O(1)。

这个方法高效地计算了所需的行数和最后一行的宽度。🚀