寫字串需要的行數

題目

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

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

示例 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個單位寬度。

注:

字串 S 的長度在 [1, 1000] 的範圍。
S 只包含小寫字母。
widths 是長度為 26的陣列。
widths[i] 值的範圍在 [2, 10]。
 

解答

思路：

1. 按字母順序編制字母數值字典
2. 編制第二個字典，行：數值
3. 得到結果。

程式碼：

class Solution:
    def numberOfLines(self, widths, S):
        """
        :type widths: List[int]
        :type S: str
        :rtype: List[int]
        """
        AP = "abcdefghijklmnopqrstuvwxyz"
        words={}
        j=0
        for i in AP:
            words[
 
i]=widths[j]
            j+=1
        
        n={}
        m=0
        j=0
        for i in S:
            m+=words[i]
            while m>100:
                n[j]=m-words[i]
                m=words[i]
                j+=1
            n[j]=m
        return [len(n),n[len(n)-1]]

結果：40%