NOIP--摆花（母函数）

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;
}