题目: Codeforces Round #258 (Div. 2)Devu and Flowers 题意:n个boxes ,第i个box有fi个flowers,每个boxes中的flowers完全相同,不同boxes的flowers不同,求从n个boxes中取出s个flowers的方案数。n<=20,s<=1e14,fi<=1e12. 排列组合的题目,一解法可用容斥原理( inclusion exclusion principle) 。 有2中写法dfs和集合。下为集合写法。
#include using namespace std;const int MOD = 1e9 + 7;typedef long long LL;LL invv[25];LL inv(LL x)/// 求逆元{ return x == 1 ? 1LL : (MOD - MOD / x) * inv (MOD % x) % MOD;}LL Cmn(LL n, LL m) ///求组合数{ LL ret = 1; for (int i = 1; i <= m; i++) ret = (n - i + 1) % MOD * ret % MOD * invv[i] % MOD; return ret;}int calc(int x){ int ret = 0; while (x) {// x -= x & (-x); x=x&(x-1); ret ^= 1; } return ret;}int n;LL s, a[25];int main(){ for (int i = 1; i < 25; i++) invv[i] = inv(i); cin >> n >> s; for (int i = 0; i < n; i++) cin >> a[i]; int ALL = (1 << n) - 1; LL ans = 0; for (int i = 0; i <= ALL; i++)///遍历所有集合 { LL nows = s; int num = 0;///统计当前集合元素个数,以确定符号 for (int j = 0; j < n; j++) { if (i & (1 << j)) { nows -= a[j] + 1; num++; } } if (nows < 0) continue; if (num & 1)///集合含有奇数个元素,符号为- ans -= Cmn(nows + n - 1, n - 1); else///集合含有偶数个元素,符号为+ ans += Cmn(nows + n - 1, n - 1); ans %= MOD; } cout << (ans + MOD) % MOD << endl; return 0;}