[BZOJ4872][Shoi2017]分手是祝愿 数学期望

先看N=k的，考虑一下怎么用最少步数解决这个问题。
一个做法是从高位到低位，没遇到一个开着的灯就将它关闭，易证这样的策略是最优的，而且是唯一的。
问题转化为目前有n个开关，其中有t个开关需要被按奇数次，求期望步数。F[t]表示由剩t个开关到剩t-1个开关所需要的期望步数。

$$F_t=\frac{t}{n}+\frac{n-t}{n}(1+F_{t+1}+F_t)$$
改写一下式子就可以得到
$$F_t = \frac{n+(n-t)F_{t+1}}{t}$$
可以在 $O(N)$的时间内处理出来
最终答案为 $\sum_{i=k+1}^tF_i$
注意把阶乘乘上去

#include <bits/stdc++.h>
#define mod 100003
#define N 1000500
using namespace std;

int n,k,a[N],inv[N],F[N],ans;

inline int rd() {
    int x=0,f=1;char ch=getchar();
    while (ch>'9'||ch<'0') {if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int get() {
    int ret = 0;
    for (int i=n;i>=1;i--) if (a[i]) {
        int tp = sqrt(i); ++ret;
        for (int j=1;j<=tp;j++) if (i%j == 0) {
            a[j] ^= 1;
            if (j*j != i) a[i/j] ^= 1;
        }
    }
    return ret;
}

int jc(int x) {
    int ret=x;
    for (int i=1;i<=n;i++) ret = 1LL * ret * i % mod;
    return ret;
}

int main() {
    n = rd(), k = rd();
    for (int i=1;i<=n;i++) a[i] = rd();
    int cur = get();
    if (cur <= k)
        printf("%d\n",jc(cur));
    else {
        inv[0] = inv[1] = 1;
        for (int i=2;i<=n;i++) inv[i] = 1LL * (mod-mod/i) * inv[mod%i] % mod;
        F[n] = 1;
        for (int i=n-1;i>=k;i--) F[i] = 1LL * (n + 1LL * (n-i) * F[i+1] % mod) * inv[i] % mod;
        for (int i=cur;i>k;i--) ans = (ans + F[i]) % mod;
        ans += k;
        printf("%d\n",jc(ans));
    }
    return 0;
}