zoukankan      html  css  js  c++  java
  • Codeforces Round #258 (Div. 2) 容斥+Lucas

    题目链接:

    http://codeforces.com/problemset/problem/451/E

    E. Devu and Flowers

    time limit per test4 seconds
    memory limit per test256 megabytes
    #### 问题描述 > Devu wants to decorate his garden with flowers. He has purchased n boxes, where the i-th box contains fi flowers. All flowers in a single box are of the same color (hence they are indistinguishable). Also, no two boxes have flowers of the same color. > > Now Devu wants to select exactly s flowers from the boxes to decorate his garden. Devu would like to know, in how many different ways can he select the flowers from each box? Since this number may be very large, he asks you to find the number modulo (109 + 7). > > Devu considers two ways different if there is at least one box from which different number of flowers are selected in these two ways.

    输入

    The first line of input contains two space-separated integers n and s (1 ≤ n ≤ 20, 0 ≤ s ≤ 1014).

    The second line contains n space-separated integers f1, f2, ... fn (0 ≤ fi ≤ 1012).

    输出

    Output a single integer — the number of ways in which Devu can select the flowers modulo (109 + 7).

    样例输入

    3 5
    1 3 2

    样例输出

    3

    题意

    给你n个花瓶,每个花瓶上装有f[i]朵颜色相同的玫瑰。现在你要从中取s朵,问有多少种取法。

    题解

    首先,如果每个花瓶里面的花都有无限朵的话,那么这就是简单的隔板问题(c[s+n-1][n-1]),那么我们可以通过容斥把问题都转换成无限制的问题。 首先我们考虑第i个超过了限制,那么相当于对剩下的sum=s-f[i]-1,做一遍无限制的隔板,这样我们对n个花瓶容斥一遍,就能求出答案。

    代码

    #include<map>
    #include<set>
    #include<cmath>
    #include<queue>
    #include<stack>
    #include<ctime>
    #include<vector>
    #include<cstdio>
    #include<string>
    #include<bitset>
    #include<cstdlib>
    #include<cstring>
    #include<iostream>
    #include<algorithm>
    #include<functional>
    using namespace std;
    #define X first
    #define Y second
    #define mkp make_pair
    #define lson (o<<1)
    #define rson ((o<<1)|1)
    #define mid (l+(r-l)/2)
    #define sz() size()
    #define pb(v) push_back(v)
    #define all(o) (o).begin(),(o).end()
    #define clr(a,v) memset(a,v,sizeof(a))
    #define bug(a) cout<<#a<<" = "<<a<<endl
    #define rep(i,a,b) for(int i=a;i<(b);i++)
    #define scf scanf
    #define prf printf
    
    typedef __int64 LL;
    typedef vector<int> VI;
    typedef pair<int,int> PII;
    typedef vector<pair<int,int> > VPII;
    
    const int INF=0x3f3f3f3f;
    const LL INFL=0x3f3f3f3f3f3f3f3fLL;
    const double eps=1e-8;
    const double PI = acos(-1.0);
    
    //start----------------------------------------------------------------------
    
    const int mod=1e9+7;
    
    LL n;
    LL s,f[22];
    
    void gcd(LL a,LL b,LL& d,LL& x,LL& y){
        if(!b){ d=a; x=1; y=0; }
        else{
            gcd(b,a%b,d,y,x);
            y-=x*(a/b);
        }
    }
    
    LL Inv(LL a,LL n){
        LL d,x,y;
        gcd(a,n,d,x,y);
        return d==1?(x+n)%n:-1;
    }
    
    LL get_C(LL n,LL m){
        LL a=1,b=1;
        for(int i=1;i<=m;i++){
            b=b*i%mod;
            a=a*(n-i+1)%mod;
        }
        return a*Inv(b,mod)%mod;
    }
    
    LL Lucas(LL n,LL m){
        if(m==0) return 1;
        return get_C(n%mod,m%mod)*Lucas(n/mod,m/mod)%mod;
    }
    
    int main() {
        scf("%I64d%I64d",&n,&s);
        for(int i=0;i<n;i++) scf("%I64d",f+i);
        LL ans=0;
        for(int i=0;i<(1<<n);i++){
            LL sum=s;
            int cnt=0;
            for(int j=0;j<n;j++){
                if((i&(1<<j))==0) continue;
                cnt++;
                sum-=(f[j]+1);
            }
            if(sum>=0){
                if(cnt&1){
                    ans-=Lucas(sum+n-1,n-1);
                }else{
                    ans+=Lucas(sum+n-1,n-1);
                }
                ans=(ans%mod+mod)%mod;
            }
        }
        prf("%I64d
    ",ans);
        return 0;
    }
    
    /*
    5 3
    1 2 9
    3 4 5
    3 5 3
    */
  • 相关阅读:
    tuntime error '55' file already open
    Visual Studio Team System 2008 Team Suite 中文版下载地址
    Windows7快捷键大全
    Thinkpad T61 on screen display 设置
    Hough变换,找直线
    多目标识别
    GALIL 数字运动控制器 应用编程指南(V1.0)
    VC调用MATLAB.
    OTSU求阈值程序
    什么是伺服电机?
  • 原文地址:https://www.cnblogs.com/fenice/p/5951019.html
Copyright © 2011-2022 走看看