【0-1背包变种】力扣2787. 将一个数字表示成幂的和的方案数

给你两个 正 整数 n 和 x 。

请你返回将 n 表示成一些 互不相同 正整数的 x 次幂之和的方案数。换句话说,你需要返回互不相同整数 [n1, n2, ..., nk] 的集合数目,满足 n = n1x + n2x + ... + nkx 。

由于答案可能非常大,请你将它对 109 + 7 取余后返回。

比方说,n = 160 且 x = 3 ,一个表示 n 的方法是 n = 23 + 33 + 53 。

示例 1:

输入:n = 10, x = 2

输出:1

解释:我们可以将 n 表示为:n = 32 + 12 = 10 。

这是唯一将 10 表达成不同整数 2 次方之和的方案。

示例 2:

输入:n = 4, x = 1

输出:2

解释:我们可以将 n 按以下方案表示:

  • n = 41 = 4 。
  • n = 31 + 11 = 4 。

提示:

1 <= n <= 300

1 <= x <= 5

cpp 复制代码
const int MOD = 1e9 + 7;
        
class Solution {
public:
    int numberOfWays(int n, int x) {
        vector<int> f(n+1);
        f[0] = 1;
        for (int i = 1; ; i++) {
            int v = std::pow(i, x);
            if (v > n) break;
            for (int s = n; s >= v; s--) {
                f[s] = (f[s] + f[s - v]) % MOD;
            }
        }
        return f[n];
    }
};

0-1背包问题的变种,首先我们定义一个一维数组f,f[s]表示可以用不同整数的 x 次幂来表示 s 的所有组合方式的数量。当 v > n 时,意味着 v 已经超过了目标数 n,所以没有必要继续计算更大的幂次,因为它们不可能再参与组合成 n。f[s] = (f[s] + f[s - v]) % MOD这个公式的意思是:当前目标数 s 的组合方式 f[s] 应该包括所有能组合成 s-v 的组合方式 f[s-v],因为你可以通过加上 v 来实现 s。最后返回 f[n],它表示用 x 次幂的整数来组合成 n 的所有不同方案的数量。

相关推荐
LNTON羚通20 分钟前
摄像机视频分析软件下载LiteAIServer视频智能分析平台玩手机打电话检测算法技术的实现
算法·目标检测·音视频·监控·视频监控
哭泣的眼泪4082 小时前
解析粗糙度仪在工业制造及材料科学和建筑工程领域的重要性
python·算法·django·virtualenv·pygame
清炒孔心菜2 小时前
每日一题 LCR 078. 合并 K 个升序链表
leetcode
Microsoft Word2 小时前
c++基础语法
开发语言·c++·算法
天才在此3 小时前
汽车加油行驶问题-动态规划算法(已在洛谷AC)
算法·动态规划
莫叫石榴姐4 小时前
数据科学与SQL:组距分组分析 | 区间分布问题
大数据·人工智能·sql·深度学习·算法·机器学习·数据挖掘
茶猫_5 小时前
力扣面试题 - 25 二进制数转字符串
c语言·算法·leetcode·职场和发展
肥猪猪爸7 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
readmancynn7 小时前
二分基本实现
数据结构·算法
萝卜兽编程7 小时前
优先级队列
c++·算法