[51nod]1778 小Q的集合[lucas定理+组合计数]

最新推荐文章于 2023-02-24 19:29:46 发布

原创最新推荐文章于 2023-02-24 19:29:46 发布 · 406 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#51nod #组合 #lucas定理

math 同时被 3 个专栏收录

8 篇文章

订阅专栏

51nod

3 篇文章

订阅专栏

dp-组合计数问题

1 篇文章

订阅专栏

本文解析了一道算法竞赛题目，探讨如何计算集合所有子集方差的期望值，通过数学推导和编程实现，利用二项式定理、逆元计算等技巧，解决了方差值可能过大的问题。

题目

小Q有一个集合 $S$ ，它的元素个数 $|S|=N$ 。

对于 $T$ 的任意一个子集合 $T$ ，定义 $f(T)=|T|^k$ ，定义 $T$ 关于 $S$ 的补集为 $S - T$ 。

小Q想知道，如果他等概率地选择一个 $S$ 的子集 $T$ ，那么 $f(T)-f(S-T)$ 的方差是多少。

由于这个方差值可能很大，不妨设其为 $v$ ，你只需要给出 $(v2^N)\%mod$ 的值即可。

题解

可以看出 $f(T)-f(S-T)$ 具有对称性,可以知道 $E(x)=0$

方差 $E[(x-E(x))^2]=\sum\binom{N}{i} (i^k-(N-i)^k)^2/2^N$ , $ans=v2^N=\sum\binom{N}{i} (i^k-(N-i)^k)^2$

但注意到 $N$ 很大则需要化解, $N,MOD$ 大小不确定,则需要 $lucas$ 定理

我们类似于 $bsgs$ 算法,对 $N$ 关于 $MOD$ 做带余除法( $lucas$ 则可以使用)

$\\ ans\\ =\sum\binom{N}{i} (i^k-(N-i)^k)^2\\ =\sum_{i=0}^{[\frac{N}{MOD}]} \sum_{j=0}^{N\%MOD} \binom{N}{i*MOD+j} ((i*MOD+j)^k-(N-(i*MOD+j))^k)^2\\ \equiv \sum_{i=0}^{[\frac{N}{MOD}]} \binom{[\frac{N}{MOD}]}{i} \sum_{j=0}^{N\%MOD} \binom{N\%MOD}{j} (j^k-(N\%MOD-j)^k)^2 \pmod {MOD}$ (这里用到了 $lucas$ 定理)

外层用二项式定理化为 $2$ 的幂,内层预处理逆元计算即可

#pragma GCC optimize(2)
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;

using LL = long long;
const int MAXN = 1e6 + 5;
LL N_MOD, NM, K, MOD;
char s[MAXN];
LL fac[MAXN], inv_fac[MAXN];
void pre_getfac(int);
LL qpow(LL, int);
LL CC(int, int);
LL ans;

int main(){
  int i;
  scanf("%s%lld%lld", s + 1, &K, &MOD);
  int len = strlen(s + 1);
  for(i = 1; i <= len; i++)
    N_MOD = 10 * N_MOD + s[i] - '0', N_MOD %= MOD;
  pre_getfac(min(N_MOD, MOD - 1));
  K %= (MOD - 1);
  int top = N_MOD / 2;
  for(i = 0; i <= top; i++){
    LL tmp = (qpow(N_MOD - i, K) - qpow(i, K) + MOD) % MOD;
    tmp = tmp * tmp % MOD;
    ans = (ans + CC(N_MOD, i) * tmp % MOD) % MOD;
  }
  ans = ans * 2 % MOD;
  LL tmp = 0;
  for(i = 1; i <= len; i++){
    tmp = 10 * tmp + s[i] - '0';
    NM = 10 * NM + tmp / MOD;
    tmp %= MOD, NM %= (MOD - 1);
  }
  ans = ans * qpow(2, NM % (MOD - 1)) % MOD;
  printf("%lld\n", ans);
  return 0;
}

LL CC(int n, int m){
  if(m > n) return 0;
  if(n < MOD) return fac[n] * inv_fac[n - m] % MOD * inv_fac[m] % MOD;
  return CC(n / MOD, m / MOD) * CC(n % MOD, m % MOD) % MOD;
}

void pre_getfac(int top){
  int i;
  fac[0] = 1;
  for(i = 1; i <= top; i++) fac[i] = fac[i - 1] * i % MOD;
  inv_fac[top] = qpow(fac[top], MOD - 2);
  for(i = top - 1; i >= 0; i--)
    inv_fac[i] = inv_fac[i + 1] * (i + 1) % MOD;
}

LL qpow(LL x, int n){
  LL res = 1;
  while(n){
    if(n & 1) res = res * x % MOD;
    x = x * x % MOD;
    n >>= 1;
  }
  return res;
}