洛谷P1077 擺花

DP 劃分類動態規劃

dp[ i ][ j ] 表示 到 第 i 種花，所有花總共取了 j 盆，總共的方案數

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <cmath>
 4 #include <cstdlib>
 5 #include <string>
 6 #include <algorithm>
 7 #include <iomanip>
 8 #include <iostream>
 9
 
 using namespace std ;
10 
11 const int maxn = 101,maxm = 101,mod = 1000007 ;
12 int n,m ;
13 int a[maxn],dp[maxn][maxm] ;
14 
15 int main() 
16 {
17     scanf("%d%d",&n,&m) ;
18     for(int i=1;i<=n;i++) 
19         scanf("%d",&a[ i ]) ;
20     dp[ 0 ][ 0 ] = 1  ;
21     for(int i=1;i<=n;i++) 
 
22         for(int j=0;j<=m;j++)    
23             for(int k=0;k<=min( j,a[i] );k++ ) 
24              dp[i][j] = (dp[i][j] + dp[i-1][j-k]) % mod ; 
25     printf("%d\n",dp[n][m]) ;
26              
27             
28     return 0 ;
29 }

洛谷P1077 擺花 動態規劃