POJ3046 动态规划多重集组合数

最新推荐文章于 2021-11-15 18:30:49 发布

acunstoppable

最新推荐文章于 2021-11-15 18:30:49 发布

阅读量189

点赞数

分类专栏：简单DP

本文链接：https://blog.csdn.net/acunstoppable/article/details/79512279

版权

简单DP 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

这篇博客探讨了POJ3046问题，涉及动态规划和多重集组合数的计算。作者强调了解决问题时前缀和优化和滚动数组优化的重要性，并提醒注意初始化及循环上下界细节的处理。文章包含解题思路和代码实现。

摘要由CSDN通过智能技术生成

题意：

蚂蚁牙黑，蚂蚁牙红：有A只蚂蚁，来自T个家族。同一个家族的蚂蚁长得一样，但是不同家族的蚂蚁牙齿颜色不同。任取n只蚂蚁(S<=n<=B)，求能组成几种集合？

思路：
1、《挑战》P69 前缀和优化；
2、滚动数组优化。
反思：
初始化以及循环上下界的细节要考虑清楚。
代码：

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;

const int MOD = 1e6;

int c[1000 + 10];
int dp[2][100000 + 10];

int main()
{
    int t, a, s, b;
    scanf("%d%d%d%d", &t, &a, &s, &b);
    for(int i = 0; i < a; i++)
    {
        int x; scanf("%d", &x);
        c[x]++;
    }
    dp[0][0] = dp[1][0] = 1;
    for(int i = 1; i <= t; i++)
    {
        for(int j = 1; j <= b; j++)
        {
            if(j <= c[i])
            {
                dp[i & 1][j] = (dp[i & 1][j - 1] + dp[(i - 1) & 1][j]) % MOD;
            }
            else
            {
                dp[i & 1][j] = (dp[(i - 1) & 1][j] + dp[i & 1][j - 1] - dp[(i - 1) & 1][j - 1 - c[i]] + MOD) % MOD;
            }
        }
    }
    int ans = 0;
    for(int j = s; j <= b; j++)
    {
        ans = (ans + dp[t & 1][j]) % MOD;
    }
    printf("%d\n", ans);
    return 0;
}