摆花 （NOIP 2012 第三题）

摆花

题目

小明的花店新开张，为了吸引顾客，他想在花店的门口摆上一排花，共 m
盆。
通过调查顾客的喜好，小明列出了顾客最喜欢的 n种花，从 1 到 n 标号。
为了在门口展出更多种花，规定第 i 种花不能超过 ai盆，摆花时同一种花放在一起，且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算，一共有多少种不同的摆花方案。
输入格式
第一行包含两个正整数 n和 m，中间用一个空格隔开。 
第二行有 n个整数，每两个整数之间用一个空格隔开，依次表示 a1,a2,…,an。
输出格式
输出只有一行，一个整数，表示有多少种方案。
注意：因为方案数可能很多，请输出方案数对 1000007
取模的结果。
数据范围
0<n,m≤100, 0≤ai≤100
输入样例：
2 4
3 2
输出样例：
2

思路分析

此考题用动态规划算法可解，时间复杂度为o(nma)

抽象出模型后，可套用多重背包问题的框子，但核心语句这里要变成累加方案数。

• 背包容量：共m盘
• 第i种花的最大盘数： ai
• 求解背包恰好摆满m盆的总方案数

动态转移方程：F[i, j] = F[i - 1, j] + F[i - 1, j - 1] + … + F[i - 1, j - a[i]]
其中：
F[i - 1, j] 代表第i种花摆0盘的可行方案数
F[i - 1, j - 1]代表第i种花摆1盘的可行方案数
…
F[i - 1, j - a[i]]代表第i种花摆a[i]盘的可行方案数

F[i，j] 的含义是前i种花恰好摆满j盘的可行方案数。
显然，F[n，m]就是题目要求解的答案。

代码实现

1. 数据结构选用二维数组

#include<iostream>
using namespace std;

//变形多重背包 统计方案数

const int maxn=105, mod = 1000007;
int n, m, a[maxn], f[maxn][maxn];
int main()
{
    cin>>n>>m;
    for(int i=1; i<=n; i++) cin>>a[i];
    f[0][0] = 1;//任意种花摆放0盘，只有一种方案： 不摆
    for(int i=1; i<=n; i++)//n种花（物品）
       for(int j=0; j<=m; j++)//总数m盘（背包容量），必须摆满
           for(int k=0; k<=min(j, a[i]); k++)//k枚举当前第i种花能放数量的各种可能性
              f[i][j] = (f[i][j] + f[i-1][j-k])%mod;//转移方程；f(i,j)初值0
    cout<<f[n][m]<<endl;
    return 0;
}

2. 因为第i种状态只依赖于第i-1种状态的结果，二维数组压缩成一维，减少空间开销。

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 110, mod = 1e6 + 7;

int n, m;
int f[N];

int main()
{
    cin >> n >> m;
    f[0] = 1;//摆0盘，一种方案
    for (int i = 1; i <=n; i ++ )//n种花
    {
        int a;
        scanf("%d",&a);
        for (int j = m; j >= 0; j -- )//必须逆序
            for (int k = 1; k<=min(j, a); k ++ )
                f[j] = (f[j] + f[j - k]) % mod;
    }

    cout << f[m] << endl;

    return 0;
}