806. 寫字符串需要的行數(shù)

我們要把給定的字符串 S 從左到右寫到每一行上，每一行的最大寬度為100個(gè)單位，如果我們?cè)趯懩硞€(gè)字母的時(shí)候會(huì)使這行超過了100 個(gè)單位，那么我們應(yīng)該把這個(gè)字母寫到下一行。我們給定了一個(gè)數(shù)組 widths ，這個(gè)數(shù)組 widths[0] 代表 'a' 需要的單位， widths[1] 代表 'b' 需要的單位，...， widths[25] 代表 'z' 需要的單位。

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

示例 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個(gè)字母，

我們需要2個(gè)整行和占用60個(gè)單位的一行。

示例 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" 將會(huì)覆蓋 9 * 10 + 2 * 4 = 98 個(gè)單位.

最后一個(gè)字母 'a' 將會(huì)被寫到第二行，因?yàn)榈谝恍兄皇Ｏ?個(gè)單位了。

所以，這個(gè)答案是2行，第二行有4個(gè)單位寬度。

嗯就是一一對(duì)應(yīng)然后判斷是否超過100就可以了

class?Solution:

????def?numberOfLines(self,?widths:?List[int],?s:?str)?->?List[int]:

????????res=0

????????line=1

????????for?tmp?in?s:

????????????n=ord(tmp)-ord('a')

????????????if?(widths[n]+res>100):

????????????????res=widths[n]

????????????????line+=1

????????????else:

????????????????res+=widths[n]

????????return?[line,res]