[NOIP2012 普及组] 摆花(背包问题)

洛谷

acwing

n种花，m个位置，每种花的摆放数量不能超过a[i]，一共有多少种摆放的方式

将花盆数量看作背包容量；
将花看作物品，体积是1，第 i种物品最多选 ai个；
将背包装满的方案数是多少
f[i,j]表示前i个物品，体积为j的方案数
f[i,j]=f[i-1,j](不选上种花)+f[i-1,j-1]（上一种花选了一朵）+f[i-1,j-2]+...

#include <iostream>
#include <bits/stdc++.h>
#include <algorithm>
#include <sstream>
#include <unordered_map>
#include <set>
#include <queue>
#include <deque>
#include <map>
#include <string>
#include <cstring>

#define x first
#define y second
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;

typedef pair<int,int> PII;
typedef pair<char,int> PCI;
typedef long long LL;
typedef unsigned long long ULL;

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


// 摆m盆花 ，有n种花，每种花摆的数量不能超过a[i],
// 同种花放在一起 可以将同种花 进行一个打包  ===> 01 背包问题 
int f[N];

void solve()
{
    int n,m;cin>>n>>m;
    f[0] = 1; // 什么花都不摆 也是一种方案 
    for(int i=0;i<n;i ++) 
    {
        int a;cin>>a;
        
        for(int j=m;j>=0;j--)
        for(int k=1;k<=a&& k <=j ;k++) //k 选 第i种 花的数量 
        f[j]=(f[j]+f[j-k] ) % mod;
    }
    
    
    
    cout<< f[m] << endl;
}


int main()
{
    ios 
    int T=1;
//     cin>>T;
    while(T -- )
    {
        solve();
    }    
    
    
    
    
    return 0;
}