BZOJ4872: [Shoi2017]分手是祝愿【几率指望DP】【思惟好题】

Description

Zeit und Raum trennen dich und mich.

时空将你我分开。B 君在玩一个游戏，这个游戏由 n 个灯和 n 个开关组成，给定这 n 个灯的初始状态，下标为从 1 到 n 的正整数。每一个灯有两个状态亮和灭，咱们用 1 来表示这个灯是亮的，用 0 表示这个灯是灭的，游戏的目标是使全部灯都灭掉。可是当操做第 i 个开关时，全部编号为 i 的约数（包括 1 和 i）的灯的状态都会被改变，即从亮变成灭，或者是从灭变成亮。B 君发现这个游戏很难，因而想到了这样的一个策略，每次等几率随机操做一个开关，直到全部灯都灭掉。这个策略须要的操做次数不少， B 君想到这样的一个优化。若是当前局面，能够经过操做小于等于 k 个开关使全部灯都灭掉，那么他将再也不随机，直接选择操做次数最小的操做方法（这个策略显然小于等于 k 步）操做这些开关。B 君想知道按照这个策略（也就是先随机操做，最后小于等于 k 步，使用操做次数最小的操做方法）的操做次数的指望。这个指望可能很大，可是 B 君发现这个指望乘以 n 的阶乘必定是整数，因此他只须要知道这个整数对 100003 取模以后的结果。

Input

第一行两个整数 n, k。

接下来一行 n 个整数，每一个整数是 0 或者 1，其中第 i 个整数表示第 i 个灯的初始状况。

1 ≤ n ≤ 100000, 0 ≤ k ≤ n；

Output

输出一行，为操做次数的指望乘以 n 的阶乘对 100003 取模以后的结果。

Sample Input

4 0
0 0 1 1

Sample Output

512

---

思路

假设没有随机

咱们考虑当前的最优算法，确定是从后往前来看有哪些须要被更新的

那么这样就很容易算出最开始咱们最少须要走几步

而后是\(dp_{i}\)表示从最少i步走到最少i-1步的指望步数

咱们考虑由于当前有i步都是须要走的，因此当前这一步转移到i-1的几率是\(\frac{i}{n}\)

而后就是走到i+1，几率\(1-\frac{i}{n}\)，指望步数能够从\(dp_{i+1}\)转移

总的转移是\(dp_{i}=\frac{i}{n}+(1-\frac{i}{n})*(1+dp_{i}+dp_{i+1})\)

移项消元获得\(dp_{i}=\frac{n+(n-i)*dp_{i+1}}{i}\)

而后就作完了，前缀和累加就好了

---

#include<bits/stdc++.h>

using namespace std;

const int Mod = 1e5 + 3;
const int N = 1e5 + 10;

int n, k, a[N], f[N], fac = 1;

int add(int a, int b) {
  return (a += b) >= Mod ? a - Mod : a;
}

int sub(int a, int b) {
  return (a -= b) < 0 ? a + Mod : a;
}

int mul(int a, int b) {
  return 1ll * a * b % Mod;
}

int fast_pow(int a, int b) {
  int res = 1;
  for (; b; b >>= 1, a = mul(a, a)) {
    if (b & 1) res = mul(res, a);
  }
  return res;
}

int main() {
  scanf("%d %d", &n, &k);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &a[i]);
    fac = mul(fac, i);
  }
  int num = 0;
  for (int i = n; i >= 1; i--) {
    for (int j = i << 1; j <= n; j += i) {
      a[i] ^= a[j];
    }
    if (a[i]) ++num;
  }
  for (int i = 1; i <= k; i++) f[i] = 1;
  for (int i = n; i > k; i--) {
    f[i] = mul(add(n, mul(n - i, f[i + 1])), fast_pow(i, Mod - 2));
  }
  int ans = 0;
  for (int i = 1; i <= num; i++) {
    ans = add(ans, f[i]);
  }
  printf("%d", mul(ans, fac));
  return 0;
}