codeforces Devu and Flowers 母函数容斥 dp

最新推荐文章于 2020-10-24 20:57:02 发布

meixiuxiudd

最新推荐文章于 2020-10-24 20:57:02 发布

阅读量517

点赞数

分类专栏： dp 数论 math 文章标签： acm codeforces dp acm 数论数学

本文链接：https://blog.csdn.net/meixiuxiudd/article/details/51695141

版权

dp 同时被 3 个专栏收录

34 篇文章 0 订阅

订阅专栏

数论

12 篇文章 0 订阅

订阅专栏

math

2 篇文章 0 订阅

订阅专栏

一种做法没见过

一种做法脑洞太大，没想到

做法1：容斥+费马小定理+lucas+壮压dp

前面都是套路，难的是壮压dp的状态

这道题倒着想

将个数为s的相同的球放入n个盒子中有多少中放法(盒子不同，盒子里面可以为空

如果我们不考虑盒子球的个数限制那么答案就是c(s+n-1,n-1)

(解释一下如果盒子可以为空的时候，我们假设当前盒子每个里面有一个，最后我们每个盒子减去一个就行了

其实就是s+n 分成n分)

但是这道题中每个盒子里面球的个数其实是有限制的

我们把盒子里面的球数超过规定的fi的情况减去

接下来就是容斥了

壮压一下。。。基本就完了。。。

中间要用到lucas 定理的原因是会爆long long

lucas 定理如下 lucas(n,m,p) = lucas(n/p,m/p,p)*c(n,m);

费马定理的应用 a/b%p = a*(b^(p-2))%p

其中的p模数都必须为素数

这道题就解决了

代码如下

/*  ^^ ====== ^^ 
ID: meixiuxiu
PROG: test
LANG: C++11
*/
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <climits>
#include <string>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <sstream>
#include <cctype>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int ,int> pii;
#define MEM(a,b) memset(a,b,sizeof a)
#define CLR(a) memset(a,0,sizeof a);
#define pi acos(-1.0)
#define maxn 40000
#define maxv 100005
const int inf = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
//#define LOCAL
ll n,s;
ll a[25];
ll quick(ll a,ll x){
    ll ans = 1;
    while(x){
        if(x&1)ans *= a,ans %= MOD;
        x >>= 1;
        a *= a;a %= MOD;
    }
    return ans;
}
ll c(ll n,ll m){
    if(n==0 || m==0)return 1;
    ll a = 1,b = 1;
    for(ll i=1;i<=m;i++){
        a *= n-i+1; a %= MOD;
        b *= i; b %= MOD;
    }
    return a*quick(b,MOD-2)%MOD;
}
ll lucas(ll n,ll m){
    ll tot = 1;
    if(n>MOD || m > MOD){
        //lucas
        tot *= c(n%MOD,m%MOD)%MOD*lucas(n/MOD,m/MOD)%MOD;
    }
    else tot *= c(n,m);
    return tot;
}
int main()
{
#ifdef LOCAL
	freopen("in.txt", "r", stdin);
//	freopen("out.txt","w",stdout);
#endif
    cin >> n >> s;
    for(int i=1;i<=n;i++)cin >> a[i];
    ll tot = lucas(s-1+n,n-1);//无限制的放
  //  cout << tot << endl;
    for(int i=1;i<(1<<n);i++){
        ll cnt = 0,sum = s;
        for(int j=0;j<n;j++){
            if((1<<j)&i)cnt++,sum -= a[j+1]+1;
        }
        if(sum<0)continue;
        if(!(cnt&1)){
            tot += lucas(sum-1+n,n-1);//盒子可以为空
            tot += MOD; tot %= MOD;
        }else{
            tot -= lucas(sum-1+n,n-1);
            tot += MOD; tot %= MOD;
        }
    }
    cout << tot << endl;
	return 0;
}

2.方法二比较数学

先要懂母函数。。这个东西推荐这个博客

http://blog.csdn.net/feeltouch/article/details/44753599

易推这个答案为xs in (1 + x + x2 + x3 + + ..xf1) * * * (1 + x + x2 + x3 + + ..xfn)中的系数