洛谷 P1077 [NOIP2012 普及组] 摆花 题解 动态规划

[NOIP2012 普及组] 摆花

题目描述

小明的花店新开张，为了吸引顾客，他想在花店的门口摆上一排花，共 $m$ 盆。通过调查顾客的喜好，小明列出了顾客最喜欢的 $n$ 种花，从 $1$ 到 $n$ 标号。为了在门口展出更多种花，规定第 $i$ 种花不能超过 $a_i$ 盆，摆花时同一种花放在一起，且不同种类的花需按标号的从小到大的顺序依次摆列。

试编程计算，一共有多少种不同的摆花方案。

输入格式

第一行包含两个正整数 $n$ 和 $m$，中间用一个空格隔开。

第二行有 $n$ 个整数，每两个整数之间用一个空格隔开，依次表示 $a_1,a_2,\cdots ,a_n$。

输出格式

一个整数，表示有多少种方案。注意：因为方案数可能很多，请输出方案数对 $10^6+7$ 取模的结果。

样例 #1

样例输入 #1

2 4
3 2

样例输出 #1

2

提示

【数据范围】

对于 $20\%$ 数据，有 $0<n\le 8,0<m\le 8,0\le a_i\le 8$。

对于 $50\%$ 数据，有 $0<n\le 20,0<m\le 20,0\le a_i\le 20$。

对于 $100\%$ 数据，有 $0<n\le 100,0<m\le 100,0\le a_i\le 100$。

NOIP 2012 普及组 第三题

原题

洛谷P1077——传送门

代码

#include <bits/stdc++.h>
using namespace std;
#define max_Heap(x) priority_queue<x, vector<x>, less<x>>
#define min_Heap(x) priority_queue<x, vector<x>, greater<x>>
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;
typedef pair<long long, long long> PLL;
const double PI = acos(-1);

const int mod = 1e6 + 7;
int a[106], dp[106];

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    // ps：这不是妥妥的分组背包吗
    // 将问题中的第i种花选k个抽象成第i组里的第k个物品
    // 就可以将问题化为分组背包方案数的计数dp
    // 最后做好题目中的细节处理即可
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    dp[0] = 1;          // 初始化一朵花不选的方案数为1
    for (int i = 1; i <= n; i++) // 枚举第i种花
    {
        for (int j = m; j >= 0; j--) // 枚举背包大小
        {
            // 由于规定第i种花不能超过ai盆，所以选取的第i种花的数量k的上界应设置为ai和背包大小的最小值
            for (int k = 1; k <= min(a[i], j); k++)
            {
                dp[j] += dp[j - k];
                dp[j] %= mod;
            }
        }
    }
    cout << dp[m];

    return 0;
}