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

最新推荐文章于 2024-07-27 17:39:19 发布

eyvr

最新推荐文章于 2024-07-27 17:39:19 发布

阅读量89

点赞数

分类专栏：用Python刷力扣文章标签： python leetcode 算法

本文链接：https://blog.csdn.net/qq_46636391/article/details/134034941

版权

用Python刷力扣专栏收录该内容

108 篇文章 1 订阅

订阅专栏

在这里插入图片描述

没什么好说的，遍历所有的 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