1. 程式人生 > >洛谷P1077 擺花(背包dp)

洛谷P1077 擺花(背包dp)

style img scan one 一個空格 space pan 不同的 pre

P1077 擺花

題目描述

小明的花店新開張,為了吸引顧客,他想在花店的門口擺上一排花,共m盆。通過調查顧客的喜好,小明列出了顧客最喜歡的n種花,從1到n標號。為了在門口展出更多種花,規定第i種花不能超過ai盆,擺花時同一種花放在一起,且不同種類的花需按標號的從小到大的順序依次擺列。

試編程計算,一共有多少種不同的擺花方案。

輸入輸出格式

輸入格式:

第一行包含兩個正整數n和m,中間用一個空格隔開。

第二行有n個整數,每兩個整數之間用一個空格隔開,依次表示a1、a2、……an。

輸出格式:

輸出只有一行,一個整數,表示有多少種方案。註意:因為方案數可能很多,請輸出方案數對1000007取模的結果。

輸入輸出樣例

輸入樣例#1:
2 4
3 2
輸出樣例#1:
2

說明

【數據範圍】

對於20%數據,有0<n≤8,0<m≤8,0≤ai≤8;

對於50%數據,有0<n≤20,0<m≤20,0≤ai≤20;

對於100%數據,有0<n≤100,0<m≤100,0≤ai≤100。

/*
多重背包 f[i]表示i盆方案數 
*/
#include<iostream>
#include<cstdio>
#include<cstring>

#define
mod 1000007 #define N 107 using namespace std; int n,m,ans,cnt; int a[N],f[N]; int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) scanf("%d",&a[i]); f[0]=1; for(int i=1;i<=n;i++) for(int j=m;j>=1;j--) for(int k=1;k<=a[i]&&k<=j;k++) f[j]
+=f[j-k]%mod,f[j]=(f[j]+mod)%mod; printf("%d\n",(f[m]+mod)%mod); return 0; }

技術分享
#include <iostream>

using namespace std;
long long b[101][101],n,m,a[101];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=0;i<=m;i++){b[i][0]=1;}
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    for(int k=j;k>=j-a[i];k--)
    {    
        if(k>=0){b[i][j]+=b[i-1][k]%1000007;b[i][j]%=1000007;}
        else break; 
    }
    cout<<b[n][m]<<endl;
}
二維

洛谷P1077 擺花(背包dp)