DP——整数划分

最新推荐文章于 2024-02-04 21:52:30 发布

莫八一

最新推荐文章于 2024-02-04 21:52:30 发布

阅读量226

点赞数

分类专栏：算法笔记文章标签：算法动态规划

本文链接：https://blog.csdn.net/mo8181/article/details/125569987

版权

算法笔记专栏收录该内容

13 篇文章 2 订阅

订阅专栏

整数划分

在这里插入图片描述

完全背包解法

整数划分问题可以看作完全背包问题，从 $1$ 到 $n$ 个数中选，正好和是 $j$ 的所有集合，每个数可以选无数个。 $f [i, j] = f [i - 1, j] + f [i - 1, j - i] + . . . + f [i - 1, j - s * i]$ , $f [i, j - i] = f [i - 1, j - i] + f [i - 1, j - 2 i] + . . . + f [i - 1, j - s * i]$ ,因此 $f [i, j] = f [i - 1, j] + f [i, j - i]$

#include <iostream>
using namespace std;
const int N = 1010, M = 1e9 + 7;

int n;
int f[N];

int main()
{
    cin >> n;
    
    f[0] = 1;
    for(int i = 1; i <= n; i ++)
        for(int j = i; j <= n; j ++)
            f[j] = (f[j] + f[j - i]) % M;
            
    cout << f[n] << endl;
    return 0;
}

思路二

我们可以考虑将使用 $i$ 个数使其总和为 $j$ ,则集合可划分为最小的数包含1和不包含1两种情况
在这里插入图片描述
对于包含1的集合我们把1去掉，等价于求 $f [i - 1, j - 1]$ ，而对于最小值不包含1的所有集合，我们让每一个都减去一个1，则等价于 $f [i - j, j]$ ，最后的转移方程为 $f [i, j] = f [i - 1, j - 1] + f [i - j, j]$ .要注意此时我们得到结果后还需要对总和是n的情况按照表示成 $1$ 到 $n$ 个数的情况求和。

#include <iostream>
using namespace std;
const int N = 1010, M = 1e9 + 7;

int n;
int f[N][N];

int main()
{
    cin >> n;
    
    f[0][0] = 1;
    
    for(int i = 1; i <= n; i ++)
        for(int j = 1; j <= i; j ++)
            f[i][j] = (f[i - 1][j - 1] + f[i - j][j]) % M;
    
    int res = 0;
    for(int i = 1; i <= n; i ++) res = (res + f[n][i]) % M;
    cout << res << endl;
    return 0;
}