[ABC292G] Count Strictly Increasing Sequences

cxqghzj發表於2024-10-05

題意

給定 \(n\) 個由 \(m\) 個字元組成的字串,你需要將裡面所有 \(?\) 替換為 \([0, 9]\),並使得字串的字典序單調遞增。

\(n, m \le 40\)

Sol

相鄰限制,想到區間 dp。

由於顯然對於同一位 \(i\)\([l, r]\) 在第 \(i\) 為上的數字一定是一段 \(0\),然後一段 \(1\),然後一段 \(2\),...,然後一段 \(9\),考慮列舉 \(k \in [l, r]\),欽定 \([l, k]\) 為當前的數字 \(j\)\([k + 1, r]\) 則由 \(j' \ge j\) 來填。

因此狀態定義十分顯然了,設 \(f_{i, j, l, r}\) 表示使用第 \(i\) 位,限制當前放的數字 \(ge j\),滿足原字串序列中 \([l, r]\) 的偏序關係的方案數。

\[f_{i, j, l, r} = \sum_k f_{i + 1, 0, l, k} \times f_{i, j + 1, k + 1, r} \]

以及:

\[f_{i, j, l, r} = f_{i, j + 1, l, r} \]

直接數位 dp 記憶化搜尋即可。

時間複雜度 \(O(n ^ 3 m c)\)

Code

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <array>
using namespace std;
#ifdef ONLINE_JUDGE

/* #define getchar() (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2) ? EOF : *p1++) */
/* char buf[1 << 23], *p1 = buf, *p2 = buf, ubuf[1 << 23], *u = ubuf; */

#endif
int read() {
    int p = 0, flg = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-') flg = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') {
        p = p * 10 + c - '0';
        c = getchar();
    }
    return p * flg;
}
void write(int x) {
    if (x < 0) {
        x = -x;
        putchar('-');
    }
    if (x > 9) {
        write(x / 10);
    }
    putchar(x % 10 + '0');
}
bool _stmer;

const int N = 45, mod = 998244353;

array <string, N> mp;
char strbuf[N];

array <array <array <array <int, N>, N>, 11>, N> f;

void Mod(int &x) {
    if (x >= mod) x -= mod;
    if (x < 0) x += mod;
}

int dfs(int x, int y, int l, int r, int m) {
    if (l > r || (l == r && x > m)) return 1;
    if (x > m || y > 9) return 0;
    if (~f[x][y][l][r]) return f[x][y][l][r];
    int ans = dfs(x, y + 1, l, r, m);
    for (int k = l; k <= r; k++) {
        if (mp[k][x] != '?' && mp[k][x] != y + '0') break;
        ans += 1ll * dfs(x + 1, 0, l, k, m) * dfs(x, y + 1, k + 1, r, m) % mod, Mod(ans);
    }
    return f[x][y][l][r] = ans;
}

bool _edmer;
int main() {
    cerr << (&_stmer - &_edmer) / 1024.0 / 1024.0 << "MB\n";
    int n = read(), m = read();

    for (int i = 0; i < N; i++)
        for (int j = 0; j < 10; j++)
            for (int k = 0; k < N; k++)
                f[i][j][k].fill(-1);

    for (int i = 1; i <= n; i++) {
        scanf("%s", strbuf);
        mp[i] = strbuf, mp[i] = " " + mp[i];
    }
    write(dfs(1, 0, 1, n, m)), puts("");
    return 0;
}

相關文章