806. 寫字符串需要的行數
我們要把給定的字符串 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]。
class Solution {
public int[] numberOfLines(int[] widths, String S) {
char[] chs = S.toCharArray();
int line = 1;
int sum = 0;
for (char c : chs) {
int width = widths[c - 'a'];
if (sum + width <= 100) {
sum += width;
} else {
line++;
sum = width;
}
}
return new int[]{line, sum};
}
}