首页 > 代码库 > 洛谷-摆花-动态规划

洛谷-摆花-动态规划

2024-10-21 15:02:02   214人阅读

题目描述

小明的花店新开张，为了吸引顾客，他想在花店的门口摆上一排花，共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。

NOIP 2012 普及组 第三题

 

思路：

　　这题是典型的动规题，我们用b[i][j]表示放第i种花，放j盆的放法。

　　首先进行初始化，无论有多少种花，如果一盆都没有，只有一种方案总数，诶这个想想嘛（动脑子噢~）

　　k是用于计算某种花放多少盆，从总盆数开始循环到（总盆数-最大盆数），如果k小于0（说明最大盆数大于总盆数）就退出循环，我们把b数组的行也就是b[i][]看成是每一种花的位置，所以我们得到状态转移方程：

b[i][j]=b[i][j]+b[i-1][k]

这个方程的意思是：我当前这个第i种花放的方法数+上一种花放k盆的方法数叠加

代码如下：

 1 #include<stdio.h>
 2 int main()
 3 {
 4     int i,j,k;
 5     long long b[1001][1001],n,m,a[1001];
 6     scanf("%d%d",&n,&m);
 7     for(i=1;i<=n;i++)
 8     {
 9         scanf("%lld",&a[i]);
10     }
11     for(i=0;i<=m;i++)
12     {
13         b[i][0]=1;//初始化，不管有多少种花，只要是0盆花，就只有1种可能性，什么都不放
14     } 
15     for(i=1;i<=n;i++)//几种花
16     {
17          for(j=1;j<=m;j++)//共放几盆花
18          {
19             for(k=j;k>=j-a[i];k--)//用k表示这种花放多少盆
20             {    
21                 if(k>=0)
22                 {
23                     b[i][j]=(b[i][j]+b[i-1][k])%1000007;
24                     b[i][j]=b[i][j]%1000007;//每次mod以防万一
25                 }
26                 else break; //如果超出限制就退出循环
27             }
28          }
29     
30     }    
31     printf("%lld\n",b[n][m]);
32 }

 

洛谷-摆花-动态规划


联系
我们