806. 写字符串需要的行数

from typing import List


class Solution:
    def numberOfLines(self, widths: List[int], s: str) -> List[int]:
        lines, cur = 0, 0
        for word in s:
            w = widths[ord(word) - ord('a')]
            if cur + w <= 100:
                cur += w
            else:
                lines += 1
                cur = w

        return [lines + (1 if cur else 0), cur]