[UOJ22]外星人(动态规划)

Description

给出一个$x$和$n$个数$ai$，你需要得到一个排列$p$，使得依次执行 $x = x \bmod a_{p_i}$后，$x$的值最大，求最后最大的 $x$ 值以及能得到这个 $x$ 这样的排列数对 $998244353$ 取模后的结果。

$n \leqslant 1000,x \leqslant5000$

Solution

可以发现这样一个性质如果$x$已经对$a_j$取模，那么对于任意未被使用的且满足$>a_j$的$a_i$，都已经无效了。

所以设$f_i$表示当前$x$的值为$i$且只考虑$\leqslant i$的$a_j$所能取得的最大值，$g_i$为 $f_i$的方案数。

那么有以下转移

$$f_i=min(f_{i \mod a_j})|a_j\leqslant i$$
$$g_i=\sum g_{i \mod a_j}\times N_i !\div N_{i \mod a_j}!$$

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const int maxn = 1005, maxm = 5005, mod = 998244353;
int n, m, a[maxn], fact[maxn], rfact[maxn], cnt[maxm], f[maxm], g[maxm];

inline int gi()
{
    char c = getchar();
    while(c < '0' || c > '9') c = getchar();
    int sum = 0;
    while('0' <= c && c <= '9') sum = sum * 10 + c - 48, c = getchar();
    return sum;
}

int main()
{
    freopen("alien.in", "r", stdin);
    freopen("alien.out", "w", stdout);

    n = gi(); m = gi();
    for(int i = 1; i <= n; ++i) a[i] = gi();
    sort(a + 1, a + n + 1);

    fact[0] = 1;
    for(int i = 1; i <= n; ++i) fact[i] = (ll)fact[i - 1] * i % mod;
    rfact[1] = 1;
    for(int i = 2; i <= n; ++i) rfact[i] = (ll)(mod - mod / i) * rfact[mod % i] % mod;
    rfact[0] = 1;
    for(int i = 2; i <= n; ++i) rfact[i] = (ll)rfact[i - 1] * rfact[i] % mod;

    f[0] = 0; g[0] = 1;
    for(int x = 1; x <= m; ++x)  {
        cnt[x] = cnt[x - 1];
        while(cnt[x] < n && a[cnt[x] + 1] <= x) ++cnt[x];
        if(cnt[x] == 0) {f[x] = x; g[x] = 1; continue;}
        f[x] = 0; g[x] = 0;
        for(int i = 1; i <= cnt[x]; ++i) 
            if(f[x] < f[x % a[i]]) f[x] = f[x % a[i]];
        for(int i = 1; i <= cnt[x]; ++i) 
            if(f[x] == f[x % a[i]]) 
                g[x] = (g[x] + (ll)g[x % a[i]] * fact[cnt[x] - 1] % mod * rfact[cnt[x % a[i]]] % mod) % mod;
    }

    printf("%d\n%lld\n", f[m], (ll)g[m] * fact[n] % mod * rfact[cnt[m]] % mod);

    return 0;
}