每日一题 2698. 求一个整数的惩罚数（中等，暴力）

没什么好说的，遍历所有的 i * i ，通过递归函数求解所有的分割方案，判断是否为惩罚数即可

class Solution:
    def punishmentNumber(self, n: int) -> int:
        def find(s, target, now):
            if now > target:
                return False
            if now + int(s) == target:
                return True
            for i in range(1, len(s)):
                if find(s[i:], target, now + int(s[:i])):
                    return True
            return False        

        ans = 0
        for i in range(1, n + 1):
            if find(str(i * i), i, 0):
                ans += i * i

        return ans
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18