1. 程式人生 > >Luogu4640 BJWC2008 王之財寶 容斥、盧卡斯定理

Luogu4640 BJWC2008 王之財寶 容斥、盧卡斯定理

範圍 pre 不能 tle 預處理 return while ons 容斥

傳送門

題意:有$N$種物品,其中$T$個物品有限定數量$B_i$,其他則沒有限定。問從中取出不超過$M$個物品的方案數,對質數$P$取模。$N,M \leq 10^9 , T \leq 15 , P \leq 10^5$


在$N$種物品中選出不超過$M$中物品的方案數可以用插板法(插板法只能滿足剛好$M$個,那麽我們可以虛構出一個數量無限的物品,把剩下的沒選擇完的都丟給它,這樣插板法就能做了)

發現$T$很小,直接容斥即可

$N,M \leq 10^9$不能直接預處理,考慮到$P$為質數且範圍較小,可以直接使用$Lucas$定理

 1 #include<bits/stdc++.h>
 2
using namespace std; 3 4 inline int read(){ 5 int a = 0; 6 bool f = 0; 7 char c = getchar(); 8 while(c != EOF && !isdigit(c)){ 9 if(c == -) 10 f = 1; 11 c = getchar(); 12 } 13 while(c != EOF && isdigit(c)){ 14 a = (a << 3
) + (a << 1) + (c ^ 0); 15 c = getchar(); 16 } 17 return f ? -a : a; 18 } 19 20 const int MAXN = 100000; 21 int N , M , P , Q , jc[MAXN + 10] , ny[MAXN + 10] , B[16] , ans; 22 23 inline int poww(long long a , int b){ 24 int times = 1; 25 while(b){ 26 if(b & 1) 27
times = times * a % Q; 28 a = a * a % Q; 29 b >>= 1; 30 } 31 return times; 32 } 33 34 inline int C(int N , int M){ 35 if(N < 0 || M < 0 || N < M) 36 return 0; 37 return 1ll * jc[N] * ny[M] % Q * ny[N - M] % Q; 38 } 39 40 int lucas(int N , int M){ 41 if(N + M == 0) 42 return 1; 43 return 1ll * C(N % Q , M % Q) * lucas(N / Q , M / Q) % Q; 44 } 45 46 void choose(int now , int num , int cnt){ 47 if(num < 0) 48 return; 49 if(now > M) 50 ans = (ans + (cnt & 1 ? -1ll : 1ll) * lucas(num + N , N) + Q) % Q; 51 else{ 52 choose(now + 1 , num , cnt); 53 choose(now + 1 , num - B[now] - 1 , cnt + 1); 54 } 55 } 56 57 int main(){ 58 #ifdef LG 59 freopen("4640.in" , "r" , stdin); 60 #endif 61 N = read(); 62 M = read(); 63 P = read(); 64 Q = read(); 65 jc[0] = ny[0] = 1; 66 for(long long i = 1 ; i < Q ; i++) 67 jc[i] = jc[i - 1] * i % Q; 68 ny[Q - 1] = poww(jc[Q - 1] , Q - 2); 69 for(long long i = Q - 2 ; i ; i--) 70 ny[i] = ny[i + 1] * (i + 1) % Q; 71 for(int i = 1 ; i <= M ; i++) 72 B[i] = read(); 73 choose(1 , P , 0); 74 cout << ans; 75 return 0; 76 }

Luogu4640 BJWC2008 王之財寶 容斥、盧卡斯定理