P8810 [蓝桥杯 2022 国 C] 数组个数 题解
阅读原文时间:2023年08月22日阅读:3

思路比较简单的一道题。

用的五维 dp,看到二维和三维的 dp 直接膜了 orz。

正文开始。

不难看出 dp。

因为 \(b_i\) 的值只与 \(a_{i-1},a_i,a_{i+1}\) 有关,所以我们定义 \(b_i\) 被 \(a_{c_1},a_{c_2},…a_{c_p}\) 满足为在这些 \(a\) 中最大的一项恰好为 \(b_i\)。

用 \(dp_{i,j,0/1,0/1,z}\) 来表示此状态下的方案数。

其中:

  • \(i\) 表示当前处于第 \(i\) 项。
  • \(j\) 表示当前这一项选择 \(j\)。
  • \(1/0\) 表示 \(b_i\) 是否被 \(a_{i-1},a_i\) 满足。
  • \(1/0\) 表示 \(b_0\) 是否被 \(a_0,a_1\) 满足。
  • \(z\) 表示 \(a_0\) 选择 \(z\)。

显然,\(a_i\) 的最大值就是 \(\min(b_{i-1},b_i,b_{i+1})\),又因为 \(b_i\le10\),所以 \(a_i\) 的最大值也是 \(10\)。

状态转移方程比较复杂。

如果 \(i>1\),那么我们需要枚举 \(a_i\) 和 \(a_{i-1}\)。

这时候存在两种情况:

  1. \(b_i\) 被 \(a_i,a_{i-1}\) 满足,此时应该更新 \(dp_{..,..,1,..,..}\) 的值。
  2. \(b_i\) 不被 \(a_i,a_{i-1}\) 满足,此时应该更新 \(dp_{..,..,0,..,..}\) 的值。

注意:在选择 \(a_i\) 的时候,\(b_{i-1}\) 必须被 \(a_{i-2},a_{i-1},a_i\) 满足,原因显然。

但当 \(i=1\) 时,应该单独讨论,因为 \(a_1\) 的选择会影响到 dp 数组的第 \(4\) 项,而 \(i\ge2\) 则不会。思路跟 \(i\ge2\) 的情况类似,不再赘述。

最后统计答案时,我们有几种情况:

  • \(dp_{n-1,x,1,1,y}\),因为此时 \(b_{n-1}\) 和 \(b_0\) 已经被满足,而其它的 \(b\) 也都被满足,直接加上即可。
  • \(dp_{n-1,x,0,1,y}\) 当 \(b_{n-1}\le y\) 时可以相加,因为 \(a\) 数组是环形的。
  • \(dp_{n-1,x,1,0,y}\) 当 \(b_0\le x\) 时可以相加,原因同上。

最后,记得取模。

臭长臭长的代码 qwq。

#include <bits/stdc++.h>

using namespace std;

const int N = 1010;
const int MOD = 1e9 + 7;

int n;
int dp[N][11][2][2][11], b[N], am[N];
int st;

inline int nxt (int idx) {
    return ((idx + 1) % n);
}

inline int pre (int idx) {
    return ((idx - 1 + n) % n);
}

int m (int x) {
    if (x < MOD) return x;
    return x - MOD;
}

int main () {
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin >> n;
    for (int i = 0;i < n;i++) cin >> b[i];
    for (int i = 0;i < n;i++) am[i] = min(b[pre(i)], min(b[i], b[nxt(i)]));
    for (int i = 0;i <= am[0];i++) if (i < b[0]) dp[0][i][0][0][i] = 1; else dp[0][i][1][1][i] = 1;
    for (int x = 0;x <= am[0];x++) for (int st = 0;st <= 1;st++) for (int j = 0;j <= am[1];j++) {
        if (j < b[1] && x < b[1]) {
            if (j < b[0]) dp[1][j][0][st][x] += m(dp[0][x][0][st][x] + dp[0][x][1][st][x]), dp[1][j][0][st][x] = m(dp[1][j][0][st][x]);
            else dp[1][j][0][1][x] += m(dp[0][x][0][st][x] + dp[0][x][1][st][x]), dp[1][j][0][1][x] = m(dp[1][j][0][1][x]);
        }
        else {
            if (j < b[0]) dp[1][j][1][st][x] += m(dp[0][x][0][st][x] + dp[0][x][1][st][x]), dp[1][j][1][st][x] = m(dp[1][j][1][st][x]);
            else dp[1][j][1][1][x] += m(dp[0][x][0][st][x] + dp[0][x][1][st][x]), dp[1][j][1][1][x] = m(dp[1][j][1][1][x]);
        }
    }
    for (int i = 2;i < n;i++) {
        int p = i - 1;
        for (int x = 0;x <= am[0];x++) for (int st = 0;st <= 1;st++) {
            for (int j = 0;j <= am[i];j++) {
                for (int z = 0;z <= am[p];z++) {
                    if (j < b[i] && z < b[i]) {
                        if (j < b[p]) dp[i][j][0][st][x] += dp[p][z][1][st][x];else dp[i][j][0][st][x] += m(dp[p][z][0][st][x] + dp[p][z][1][st][x]);
                        dp[i][j][0][st][x] = m(dp[i][j][0][st][x]);
                    }
                    else {
                        if (j < b[p]) dp[i][j][1][st][x] += dp[p][z][1][st][x];
                        else dp[i][j][1][st][x] += m(dp[p][z][0][st][x] + dp[p][z][1][st][x]);
                        dp[i][j][1][st][x] = m(dp[i][j][1][st][x]);
                    }
                }
            }
        }
    }
    int ans = 0;
    for (int x = 0;x <= am[0];x++) for (int i = 0;i <= am[n - 1];i++) {
        ans += dp[n - 1][i][1][1][x]; ans = m(ans);
        if (i >= b[0]) ans += dp[n - 1][i][1][0][x];
        if (x >= b[n - 1]) ans += dp[n - 1][i][0][1][x];
        ans = m(ans);
    }
    cout << ans << endl;
    return 0;
}