我們要把給定的字串 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]。

import java.util.*;

public class Main {

    public static void main(String[] args)
    {

        Scanner sc=new Scanner(System.in);
       
        int arr[]={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};

        String str="bbbcccdddaaa";
        Solution.numberOfLines(arr,str);

    }
}


class Solution {
    public static int[] numberOfLines(int[] widths, String S) {

        int len=0;                       //記錄單行長度
        int hang=1;                    //記錄行數
        for (int i=0;i<S.length();i++)
        {
            len+=widths[S.charAt(i)-97];
            if (len>100)                //當長度超過100時，換行，行數加1，長度從頭開始計算
            {
                hang++;
                len=widths[S.charAt(i)-97];
            }
        }
        int arr[]=new int [2];
        arr[0]=hang;
        arr[1]=len;
        return arr;
    }
}
 

思路：記錄字元需要單位的陣列對比a-z來說是一一對應的，字元的ASCII碼-97剛好與下標對應，可以用這個規律獲取長度，當超過100單位的時候，行數應該加一，然後重頭計算長度。