给你一个由若干数字(0 - 9)组成的字符串 s ,和一个整数。
如果 s 的长度大于 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) 链接:https://leetcode-cn.com/problems/calculate-digit-sum-of-a-string 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
import math
class Solution:
def digitSum(self, s: str, k: int) -> str:
while len(s)>k:
n = math.ceil(len(s)/k)
res = [s[i*k : (i+1)*k] for i in range(n)] # 切出子串
res = [[ord(c)-ord('0') for c in si] for si in res ] # 子串的数字
res = [str(sum(x)) for x in res] # 子串数字的和
s = ''.join(res) # 合并成字符串
return s
28 ms 14.9 MB Python3