首页 > 代码库 > [BestCoder Round#26] Apple 【组合数学】
[BestCoder Round#26] Apple 【组合数学】
题目链接:HDOJ - 5160
题目分析
第一眼看上去,要求统计所有不同排列对答案的贡献。嗯...完全没有想法。
但是,如果我们对每个数字单独考虑,计算这个数字在总答案中的贡献,就容易多了。
对于一个数字 ai ,有 ni 个 。比它大的数字有 p 个,比它小的数字有 q 个。所有的数字一共有 N 个。
首先,比它小的数字对它不会造成影响,所以我们只要考虑它和比它大的数字。那么我们就在 N 个位置中,选 (ni + p) 个位置,给它和比它大的数字。
然后比它大的数字有 x1 种排列,比它小的数字有 x2 种排列。这个如何来求呢?这是多重集排列。
多重集排列,对于一个多重集 A={a1*n1, a2*n2, a3*n3, ak*nk} 。排列数为 Sum(n1...nk)!/(n1!n2!n3!...nk!)
那么 ai 有多少个会被记入答案呢?我们枚举每种情况:有 ni 个记入答案, (ni-1) 个, (ni-2)个 .... 1 个。
如果有 k 个 ai 无法记入答案,说明有比它们大的数在他们前面,也就是说他们被插到了 p 个比他们大的数后面或之间。这 p 个数后面一共有 p 个位置,每个位置都可以插入任意个 ai ,这种情况数为 C(k + p - 1, k) (相当于向 p 个箱子里分配 k 个球,隔板法)。
那么 ai 对答案的贡献就为 C(N, ni + p) * x1 * x2 * sigma((ni - k) * C(k + p - 1, k)) (0 <= k < ni) * ai 。
Warning!
出现的错误: % 的优先级比 + 高!如果这样 Ans = Ans + temp % Mod 。Ans 是会爆掉的!!应该是 Ans = (Ans + Temp) % Mod 。
昨天晚上因为这个错误debug了3hours!!!!!!
代码
#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>#include <cmath>#include <algorithm>using namespace std;const int MaxN = 100000 + 15; typedef long long LL;const LL Mod = 1000000007ll;int T, n, Top;LL Ans;LL A[MaxN], f[MaxN], g[MaxN], Sum[MaxN], NSum[MaxN], Fac[MaxN], NY_Fac[MaxN];struct ES_Num{ LL Cnt, Num; ES_Num() {} ES_Num(LL a, LL b) { Num = a; Cnt = b; }} ES[MaxN];LL Pow(LL a, LL b) { LL f, ret; f = a; ret = 1ll; while (b) { if (b & 1) { ret *= f; ret %= Mod; } b >>= 1; f *= f; f %= Mod; } return ret;}void Init() { int Max_Num = 100000 + 5; Fac[0] = 1; for (int i = 1; i <= Max_Num; ++i) Fac[i] = Fac[i - 1] * i % Mod; for (int i = 0; i <= Max_Num; ++i) NY_Fac[i] = Pow(Fac[i], Mod - 2);}LL C(LL a, LL b) { if (b == 0) return 1; if (a < b) return 0; LL ret; ret = Fac[a]; ret = ret * NY_Fac[a - b] % Mod; ret = ret * NY_Fac[b] % Mod; return ret;}int main() { Init(); scanf("%d", &T); for (int Case = 1; Case <= T; ++Case) { scanf("%d", &n); Ans = 0; for (int i = 1; i <= n; ++i) scanf("%d", &A[i]); sort(A + 1, A + n + 1); Top = 0; for (int i = 1; i <= n; ++i) { if (i != 1 && A[i] == A[i - 1]) ++(ES[Top].Cnt); else ES[++Top] = ES_Num((LL)A[i], 1); } Sum[0] = 0; for (int i = 1; i <= Top; ++i) Sum[i] = Sum[i - 1] + ES[i].Cnt; NSum[Top + 1] = 0; for (int i = Top; i >= 1; --i) NSum[i] = NSum[i + 1] + ES[i].Cnt; LL t = 1; f[0] = 1; for (int i = 1; i <= Top; ++i) { t = t * Fac[Sum[i]] % Mod * NY_Fac[ES[i].Cnt] % Mod; f[i] = t; t = t * NY_Fac[Sum[i]] % Mod; } t = 1; g[Top + 1] = 1; for (int i = Top; i >= 1; --i) { t = t * Fac[NSum[i]] % Mod * NY_Fac[ES[i].Cnt] % Mod; g[i] = t; t = t * NY_Fac[NSum[i]] % Mod; } LL x, y; for (int i = 1; i <= Top; ++i) { x = 0; for (int j = 0; j < ES[i].Cnt; ++j) { y = (ES[i].Cnt - j) * C(NSum[i + 1] + j - 1, j) % Mod; x = (x + y) % Mod; } x = x * ES[i].Num % Mod; x = x * f[i - 1] % Mod; x = x * g[i + 1] % Mod; x = x * C(NSum[1], NSum[i]) % Mod; Ans = (Ans + x) % Mod; } printf("Case #%d: %d\n", Case, (int)Ans); } return 0;}
[BestCoder Round#26] Apple 【组合数学】