題目

難度：★☆☆☆☆
類型：字符串

我們要把給定的字符串 S 從左到右寫到每一行上，每一行的最大寬度為100個單位，如果我們在寫某個字母的時候會使這行超過了100 個單位，那么我們應該把這個字母寫到下一行。我們給定了一個數組 widths ，這個數組 widths[0] 代表 'a' 需要的單位， widths[1] 代表 'b' 需要的單位，...， widths[25] 代表 'z' 需要的單位。

現在回答兩個問題：至少多少行能放下S，以及最后一行使用的寬度是多少個單位？將你的答案作為長度為2的整數列表返回。

注意
字符串 S 的長度在 [1, 1000] 的范圍。
S 只包含小寫字母。
widths 是長度為 26的數組。
widths[i] 值的范圍在 [2, 10]。

示例

示例 1
輸入:
widths = [10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10]
S = "abcdefghijklmnopqrstuvwxyz"
輸出: [3, 60]
解釋:
所有的字符擁有相同的占用單位10。所以書寫所有的26個字母，
我們需要2個整行和占用60個單位的一行。

示例 2
輸入:
widths = [4,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10]
S = "bbbcccdddaaa"
輸出: [2, 4]
解釋:
除去字母'a'所有的字符都是相同的單位10，并且字符串 "bbbcccdddaa" 將會覆蓋 9 * 10 + 2 * 4 = 98 個單位.
最后一個字母 'a' 將會被寫到第二行，因為第一行只剩下2個單位了。
所以，這個答案是2行，第二行有4個單位寬度。

解答

這里需要注意：

width數組的長度是26，我們可以通過該數組查詢各個字母需要的空間；

如果一行中添加一個字母所占的空間不大于100，則可以添加，否則需要換行。

我們這里用line_num記錄行數，用line記錄最后一行的已占用空間，每次出現換行操作，line_num+1，并且更新最后一層空間line。

class Solution:
    def numberOfLines(self, widths, S):
        line_num = 1
        line = 0
        for i in S:
            num = widths[ord(i)-97]
            if line + num > 100:
                line_num += 1
                line = num
            else:
                line += num
        return [line_num, line]

如有疑問或建議，歡迎評論區留言~