1778: 摆花
时间限制: 1 Sec 内存限制: 128 MB提交: 128 解决: 48
[ 提交][ 状态][ 讨论版][命题人: admin]
题目描述
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
试编程计算,一共有多少种不同的摆花方案。
输入
第一行包含两个正整数n和m,中间用一个空格隔开。
第二行有n个整数,每两个整数之间用一个空格隔开,依次表示a1、a2、……an。
0<n≤100,0<m≤100,0≤ ai≤100
输出
输出只有一行,一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对1000007取模的结果。
样例输入
2 4
3 2
样例输出
2
题解:很久以前也经历过一道HDU类似的题目“查单词”,被折磨良久,这次比赛再次遇到这道题期初也是很懵逼的,努力回想当时考虑的马马虎虎的思路,结果屡次中断,只能拿出记在笔记上面的母函数模板,实在是弱。只得回去再看一遍母函数的应用,出乎意料的居然清晰了好多。
母函数即为模拟多项式相乘,最终结果的指数是取得多少的价值,系数为拿到该价值的方案数。则可建立两个数组C1和C2,一个为模拟下一个要乘进去的多项式,一个是累乘出的多项式,(刚刚在考虑的时候居然天真的以为可以直接用一个数组,后来发现是自己傻了(⊙_⊙)?),三个for循环,第一个是乘的次数,第二个为我累乘出的多项式的指数,第三个为我要乘进去的多项式的指数,将指数相同的系数加起来存到c2,再赋值给c1,同时清空c2(一定要清空,否则可能使用c1中累乘的值)最终数组c1即为拿到各价值的方案数。
#include<stdio.h>
#include <algorithm>
#include<iostream>
#include<string.h>
#include<vector>
#include<stdlib.h>
#include<math.h>
#include<queue>
#include<deque>
#include<ctype.h>
#include<map>
#include<set>
#include<stack>
#include<string>
#include<algorithm>
#define ll long long
#define INF 0x3f3f3f3f
#define FAST_IO ios::sync_with_stdio(false)
const double PI = acos(-1.0);
const double eps = 1e-6;
using namespace std;
int a[1005],c1[1005],c2[1005];
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=0;i<=a[1];i++)
c1[i]=1;
for(int i=2;i<=n;i++)
{
for(int j=0;j<=m;j++)
for(int k=0;j+k<=m && k<=a[i];k++)
c2[j+k]+=c1[j],c2[j+k]%=1000007;
memcpy(c1,c2,sizeof(c2));
memset(c2,0,sizeof(c2));
}
cout<<c1[m]<<endl;
return 0;
}