LeetCode簡單題之計算字符串的數字和

題目

給你一個由若幹數字（0 - 9）組成的字符串 s ，和一個整數。
如果 s 的長度大於 k ，則可以執行一輪操作。在一輪操作中，需要完成以下工作：
將 s 拆分 成長度為 k 的若幹 連續數字組 ，使得前 k 個字符都分在第一組，接下來的 k 個字符都分在第二組，依此類推。注意，最後一個數字組的長度可以小於 k 。
用錶示每個數字組中所有數字之和的字符串來 替換 對應的數字組。例如，“346” 會替換為 “13” ，因為 3 + 4 + 6 = 13 。
合並 所有組以形成一個新字符串。如果新字符串的長度大於 k 則重複第一步。
返回在完成所有輪操作後的 s 。
示例 1：
輸入：s = “11111222223”, k = 3
輸出：“135”
解釋：

• 第一輪，將 s 分成：“111”、“112”、“222” 和 “23” 。
  接著，計算每一組的數字和：1 + 1 + 1 = 3、1 + 1 + 2 = 4、2 + 2 + 2 = 6 和 2 + 3 = 5 。
  這樣，s 在第一輪之後變成 “3” + “4” + “6” + “5” = “3465” 。
• 第二輪，將 s 分成：“346” 和 “5” 。
  接著，計算每一組的數字和：3 + 4 + 6 = 13 、5 = 5 。
  這樣，s 在第二輪之後變成 “13” + “5” = “135” 。
  現在，s.length <= k ，所以返回 “135” 作為答案。
  示例 2：
  輸入：s = “00000000”, k = 3
  輸出：“000”
  解釋：
  將 “000”, “000”, and “00”.
  接著，計算每一組的數字和：0 + 0 + 0 = 0 、0 + 0 + 0 = 0 和 0 + 0 = 0 。
  s 變為 “0” + “0” + “0” = “000” ，其長度等於 k ，所以返回 “000” 。
  提示：
  1 <= s.length <= 100
  2 <= k <= 100
  s 僅由數字（0 - 9）組成。
  來源：力扣（LeetCode）

解題思路

  這樣的題目一看就非常能想到遞歸來試下。一眼就能看出可以遞歸的題目往往寫法上比較簡單，只需要符合第一個結果就能馬上寫出code。在每一層遞歸裏做一個簡單的事，就是將整個字符串先三個三個按照要求拼成一個新的字符串，至於新的字符串長度是什麼情况無需關注，下一層自會解决。

class Solution:
    def digitSum(self, s: str, k: int) -> str:
        if len(s)<=k:  #遞歸出口
            return s
        S,temp,count='','',0
        for i in s:
            temp+=i
            count+=1
            if count%k==0:
                S+=str(sum([int(j) for j in temp]))
                temp=''
        if temp:  #假如s的長度不是k的整數倍，處理尾巴
            S+=str(sum([int(j) for j in temp]))
        s=S
        return self.digitSum(s,k)