洛谷 P1077 摆花

洛谷 P1077 摆花

题目原地址 https://www.luogu.org/problem/P1077

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

2 4
3 2

-输出

2

• 数据范围
  对于20%数据，有0< n ≤8 , 0< m ≤ 8 , 0 ≤ a_i ≤ 8 ;
  对于50%数据，有0 < n ≤ 20 , 0 < m ≤ 20 , 0 ≤ a_i ≤ 20 ;
  对于100%数据，有0 < n ≤ 100 , 0 < m ≤ 100 , 0 ≤ a_i ≤ 100 ;

---

首先，我们想到的一定是暴力dfs—20分

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,ans,a[105],b[105];
void dfs(int k,int space){
    if(!space){
        ans++;
        ans%=1000007;
        return;
    }
    if(k==n)return;
    for(int i=min(a[k],space);space-i<=b[k]&&i>=0;i--){
        if(space-i<b[k])dfs(k+1,space-i);
        else ans++;
    }
    ans%=1000007;
    return;
}
int main(){
//   freopen("flower.in","r",stdin);
//   freopen("flower.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)scanf("%d",a+i);
    for(int i=n-1;i>=0;i--)b[i]=b[i+1]+a[i+1];
    dfs(0,m);
    printf("%d",ans);
//   fclose(stdin);
//   fclose(stdout);
    return 0;
}

---

剪个枝可能就到50分了

但是，数据量告诉我们，这道题必须用dp

dp[i][j]表示只用前 i 种花，共摆 m 盆的方法数量
那么，我们可以知道，dp[i][j]=dp[i-1][j-1]+dp[i-1][j-2]+dp[i-1][j-3]+…+dp[i-1][j-a[i]+1]+dp[i-1][j-a[i]]
看不懂就多看几遍
那么，动态转移可以用3层循环解决
_{我绝对不会告诉你dp数组是可以滚动的}

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,ans,a[105],dp[1000000];
int main(){
//   freopen("flower.in","r",stdin);
//   freopen("flower.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)scanf("%d",a+i);
    tot[0]=a[0];
    for(int i=0;i<=a[0];i++)dp[i]=1;
    for(int i=1;i<n;i++){
        for(int j=m;j>0;j--)
           for(int k=1;k<=a[i]&&j>=k;k++){
               dp[j]+=dp[j-k];
               dp[j]%=1000007;
        }
    }
    printf("%d",dp[m]);
//   fclose(stdin);
//   fclose(stdout);
    return 0;
}