題目
難度：★☆☆☆☆
類型：字符串
我們要把給定的字符串 S 從左到右寫到每一行上，每一行的最大寬度為100個(gè)單位和媳，如果我們?cè)趯懩硞€(gè)字母的時(shí)候會(huì)使這行超過(guò)了100 個(gè)單位，那么我們應(yīng)該把這個(gè)字母寫到下一行。我們給定了一個(gè)數(shù)組 widths 沃呢，這個(gè)數(shù)組 widths[0] 代表 'a' 需要的單位断序， widths[1] 代表 'b' 需要的單位流纹，...， widths[25] 代表 'z' 需要的單位违诗。
現(xiàn)在回答兩個(gè)問(wèn)題：至少多少行能放下S漱凝，以及最后一行使用的寬度是多少個(gè)單位？將你的答案作為長(zhǎng)度為2的整數(shù)列表返回诸迟。
注意
字符串 S 的長(zhǎng)度在 [1, 1000] 的范圍茸炒。
S 只包含小寫字母。
widths 是長(zhǎng)度為 26的數(shù)組阵苇。
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個(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è)單位寬度关斜。
解答
這里需要注意：
width數(shù)組的長(zhǎng)度是26，我們可以通過(guò)該數(shù)組查詢各個(gè)字母需要的空間铺浇；
如果一行中添加一個(gè)字母所占的空間不大于100痢畜，則可以添加，否則需要換行鳍侣。
我們這里用line_num記錄行數(shù)丁稀，用line記錄最后一行的已占用空間，每次出現(xiàn)換行操作倚聚，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]
如有疑問(wèn)或建議惑折，歡迎評(píng)論區(qū)留言~
