洛谷 P1077 [NOIP2012 普及组] 摆花

最新推荐文章于 2024-10-02 22:45:05 发布

是小Y啦

最新推荐文章于 2024-10-02 22:45:05 发布

阅读量230

点赞数 2

文章标签：算法动态规划 c++

本文链接：https://blog.csdn.net/m0_73917165/article/details/138167340

版权

本文介绍了一种使用线性动态规划解决的问题，计算在给定花的种类和数量限制下，有多少种不同的摆放方案。通过设置dp数组并定义转移方程，模拟了每种花的可能选择，最终得出总方案数。

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

思路：线性DP。

这道题我们直接就可以找到动态方程的状态，那就是对花的种类，摆放花的数量来讲。

我们直接设dp[i][j]表示为i种花时摆放j个花的方案数。

这样的话，我们可以这样想：首先dp[0][0]肯定是1，因为这个时候没有种类也没有摆放花，所以只有这一种方案数；后来，当我们每多一种花的种类时，我们需要继承上一个没有多当前花种的状态，也就是说，我们需要加上上一个状态的方案数才能保证这个状态的方案数。

模拟来说，也就是对于每一种花，我们都选取一定的数量，这个数量需要我们进行枚举，然后算出来方案数，最后加一块统计。

转移方程就不多说了，因为和背包问题比较像：

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<cstdlib>
#include<cmath> 
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<deque>
#include <iomanip>
#include<sstream>
#include<numeric>
#include<map>
#include<limits.h>
#include<unordered_map>
#include<set>
#define int long long
#define MAX 110
#define inf 0x3f3f3f3f
#define _for(i,a,b) for(int i=a;i<(b);i++)
using namespace std;
typedef pair<int, int> PII;
int n, m;
int counts;
int dx[] = { 0,1,0,-1};
int dy[] = { 1,0,-1,0 };
int dp[MAX][MAX];
int arr[MAX];
int mod = 1e6 + 7;
signed main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    cin >> n >> m;
    dp[0][0] = 1;
    for (int i = 1; i <= n; i++) {
        cin >> arr[i];
    }
    for (int i = 1; i <= n; i++) {//花的种类
        for (int j = 0; j <= m; j++) {//摆放花的数量
            for (int k = 0; k <= min(arr[i], j); k++) {//min是为了判断当前摆放的花和种类种的花的个数哪一个小，因为并不能摆放超过当前个数的数量
                dp[i][j] = (dp[i][j] + dp[i - 1][j - k])%mod;//转移方程，不要忘记取余。            }
        }
    }
    cout << dp[n][m] << endl;
    return 0;
}