Leetcode.2698 求一个整数的惩罚数

题目链接

Leetcode.2698 求一个整数的惩罚数 rating : 1679

题目描述

给你一个正整数 n n n ,请你返回 n n n 的 惩罚数

n n n 的 惩罚数 定义为所有满足以下条件 i i i 的数的平方和:

  • 1 ≤ i ≤ n 1 \leq i \leq n 1≤i≤n
  • i ∗ i i * i i∗i 的十进制表示的字符串可以分割成若干连续子字符串,且这些子字符串对应的整数值之和等于 i i i 。
示例 1:

输入:n = 10

输出:182

解释:总共有 3 个整数 i 满足要求:

  • 1 ,因为 1 * 1 = 1
  • 9 ,因为 9 * 9 = 81 ,且 81 可以分割成 8 + 1 。
  • 10 ,因为 10 * 10 = 100 ,且 100 可以分割成 10 + 0 。 因此,10 的惩罚数为 1 + 81 + 100 = 182
示例 2:

输入:n = 37

输出:1478

解释:总共有 4 个整数 i 满足要求:

  • 1 ,因为 1 * 1 = 1
  • 9 ,因为 9 * 9 = 81 ,且 81 可以分割成 8 + 1 。
  • 10 ,因为 10 * 10 = 100 ,且 100 可以分割成 10 + 0 。
  • 36 ,因为 36 * 36 = 1296 ,且 1296 可以分割成 1 + 29 + 6 。 因此,37 的惩罚数为 1 + 81 + 100 + 1296 = 1478
提示:
  • 1 ≤ n ≤ 1000 1 \leq n \leq 1000 1≤n≤1000

解法:回溯

我们定义 d f s ( u , s u m , t , s ) dfs(u,sum,t,s) dfs(u,sum,t,s) 表示 s s s 能否拆分成若个子字符串,能够满足这些子字符串的值加起来 = t = t =t。

我们直接回溯枚举每一个子串的分割位置,求出所有可能。

时间复杂度: O ( n 1 + 2 log ⁡ 2 10 ) O(n^{1 + 2 \log_{2}^{10}}) O(n1+2log210) , n n n 是给定的元素。对于给定的元素 n 2 n^2 n2,将其转换为字符串的长度为 ⌊ m = 1 + 2 log ⁡ 10 i ⌋ \lfloor m = 1 + 2 \log_{10}^{i} \rfloor ⌊m=1+2log10i⌋,回溯时的子状态为 2 m 2^m 2m 个,所以时间复杂度为 O ( n 1 + 2 log ⁡ 2 10 ) O(n^{1 + 2 \log_{2}^{10}}) O(n1+2log210)。

C++代码:

cpp 复制代码
class Solution {
public:
    int punishmentNumber(int n) {
        int ans = 0;

        function<bool(int,int,int,string&)> dfs = [&](int u,int sum,int t,string& s)->bool{
            if(u >= s.size()){
                return sum == t;
            }
            if(sum > t) return false;
            
            for(int i = u , d = 0;i < s.size();i++){
                d = d * 10 + s[i] - '0';
                if(dfs(i + 1,sum + d,t,s)) return true;
            }

            return false;
        };

        for(int x = 1;x <= n;x++){
            string s = to_string(x * x);
            if(dfs(0,0,x,s)) ans += x * x;
        }

        return ans;
    }
};
相关推荐
丶Darling.2 天前
代码随想录 | Day38 | 动态规划 :01背包应用 目标和&&一和零
c++·算法·动态规划·记忆化搜索·回溯·递推·01背包
银氨溶液5 天前
力扣——113. 路径总和
数据结构·算法·leetcode·职场和发展·dfs·回溯
IronmanJay1 个月前
【LeetCode每日一题】——679.24 点游戏
数据结构·算法·leetcode·回溯·递归·679.24 点游戏·24点游戏
Y_3_71 个月前
【回溯数独】有效的数独(medium)& 解数独(hard)
java·数据结构·windows·算法·dfs·回溯
CXDNW1 个月前
【算法篇】回溯算法类(1)(笔记)
c++·笔记·算法·leetcode·回溯·递归
IronmanJay1 个月前
【LeetCode每日一题】——95.不同的二叉搜索树 II
数据结构·算法·leetcode·回溯·递归·二叉搜索树·95.不同的二叉搜索树 ii
IronmanJay1 个月前
【LeetCode每日一题】——LCP 51.烹饪料理
数据结构·算法·leetcode·回溯·数组·递归·lcp 51.烹饪料理
山顶夕景2 个月前
【Leetcode152】分割回文串(回溯 | 递归)
算法·深度优先·回溯
志远19972 个月前
算法笔试-编程练习-H-02-24
算法·模拟·回溯·大厂·最大团
逝去的秋风3 个月前
【代码随想录训练营第42期 Day23打卡 回溯Part2 - LeetCode 39. 组合总和 40.组合总和II 131.分割回文串
leetcode·剪枝·回溯·递归