2020ICPC上海站 E题 The Journey of Geor Autumn 组合数学 + dp

传送门： https://ac.nowcoder.com/acm/contest/9925/E

题意

$有多少个满足1...n的全排列中，i>k时，a_i>mi{n}_{j\in [i-k,i)}{a}_j。(n\ge 10^{7}k\ge 10^7)$

思路

$[1，n]的区间里，因为i>k时才会有要求，所以将最小的数放在[1,k]当中。$

$设位置为pos，则$

• $[1,pos)的数随便放，有C_{n-1}^{pos-1}\ast (pos-1)!种方案$
• $(pos,n]的方法就又要要求了，假设方案数为f[n-pos]$

$所以我们得出$

$$f[n]=\sum _{j=1}^{min(n,k)}{C}_{n-1}^{j-1}\ast (j-1)!\ast f[n-j]$$
$这样我们就要递归出子问题了，即：$

$$f[i]=\sum _{j=1}^{min(i,k)}{C}_{i-1}^{j-1}\ast (j-1)!\ast f[i-j]$$

$所以暴力O(nk)的状态转移的话，就应该这样：$

void zhuanyi() {
    for(int i = 1;i <= n; i++) {
        for(int j = 1;j <= min(i, k); j++) {
            f[i] += f[i - j] * C(i - 1, j - 1) * fac[j - 1];
        }
    }
}

$但是O(nk)是1e14的复杂度，这远远超标了，所以我们要优化以下，将转移方程优化为：$

$$f[i]=(i-1)!\ast \sum _{j=1}^{min(i,k)}\frac{f[i-j]}{(i-j)!}$$

$可以看到，等式右边就是一个前缀和，所以转移的过程中维护前缀和即可。$

$所以我们先预处理阶乘和阶乘逆元，在前缀和优化即可。$

Code（345MS）

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<int, int> pdd;

#define INF 0x3f3f3f3f
#define lowbit(x) x & (-x)
#define mem(a, b) memset(a , b , sizeof(a))
#define FOR(i, x, n) for(int i = x;i <= n; i++)

 const ll mod = 998244353;
// const ll mod = 1e9 + 7;
// const double eps = 1e-6;
// const double PI = acos(-1);
// const double R = 0.57721566490153286060651209;

const int N = 1e7 + 10;

ll quick_pow(ll a, ll b) {
    ll ans = 1;
    while(b) {
        if(b & 1) ans = ans * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ans % mod;
}

ll fac[N], invfac[N];

void init() {
    fac[0] = 1;
    for(int i = 1;i < N; i++) {
        fac[i] = fac[i - 1] * i % mod;
    }
    invfac[N - 1] = quick_pow(fac[N - 1], mod - 2);
    for(int i = N - 1;i >= 1; i--) {
        invfac[i - 1] = invfac[i] * i % mod;
   }
}

ll f[N];
ll sum[N];

void solve() {
    init();
    int n, k;
     cin >> n >> k;
     f[0] = sum[0] = 1;
     for(int i = 1;i <= n; i++) {
         f[i] = (sum[i - 1] - (i - k - 1 >= 0 ? sum[i - k - 1] : 0)) * fac[i - 1] % mod;
         sum[i] = (sum[i - 1] + f[i] * invfac[i] % mod) % mod;
     }
     cout << (f[n] % mod + mod) % mod << endl;
}

signed main() {
    solve();
}