洛谷P1077题(2012年NOIP普及组):用动态规划解决摆花问题
一、题目解读
洛谷P1077摆花问题考察动态规划的实际应用。题目要求用n种花摆出m盆花束,每种花有数量限制a[i],需要计算所有可能的摆放方案数。这个问题在算法竞赛中具有典型性,能帮助学习者掌握多重背包问题的变种解法。
二、解题思路
问题转化:将每种花看作背包问题中的物品,花盆数量对应背包容量
状态定义:dp[i][j]表示前i种花摆放j盆的方案数
转移方程:通过三重循环实现状态转移,考虑每种花摆放0到a[i]盆的所有可能
模数处理:使用1e6+7作为模数避免整数溢出
三、解题步骤
外层循环遍历花种类(1-n)
中层循环遍历目标花盆数(1-m)
内层循环枚举当前种类花的摆放数量(0-min(a[i],j))
状态转移:dp[i][j] += dp[i-1][j-k]
最终输出dp[n][m]%MOD的结果
四、完整代码
#include <iostream> #include <vector> using namespace std; const int MOD = 1e6 + 7; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int n, m; cin >> n >> m; vector<int> a(n + 1); // 花的数量限制,从1开始编号 for (int i = 1; i <= n; ++i) { cin >> a[i]; } // dp[i][j]表示前i种花摆放j盆的方案数 vector<vector<int>> dp(n + 1, vector<int>(m + 1, 0)); // 初始化:前i种花摆放0盆的方案数为1(什么都不放) for (int i = 0; i <= n; ++i) { dp[i][0] = 1; } for (int i = 1; i <= n; ++i) { for (int j = 1; j <= m; ++j) { // 第i种花可以放0到min(a[i],j)盆 for (int k = 0; k <= a[i] && k <= j; ++k) { dp[i][j] = (dp[i][j] + dp[i - 1][j - k]) % MOD; } } } cout << dp[n][m] << endl; return 0; }
五、总结
本题展示了动态规划解决组合计数问题的典型模式。关键点在于:
原创内容 转载请注明出处