场上死活不会,下来再看一下就会了,两周没开电脑导致的。
首先考虑如何计算 \(f_S(p)\)。
首先因为 \(f_S(p)\) 的限制都是对于前缀的图的限制,所以先来考察前缀的图的结构和变化情况。
经过手玩能够知道,对于前 \(i\) 个数的图,根据值域划分,连通块就为一些相邻的区间 。
然后在最后加入了一个数 \(a_{i + 1} = x\)(此时考虑的是相对大小)后,就相当于是加入了 \([x, x + 1)\) 这个区间,并且把区间右端点大于 \(x\) 的区间都合并到一起。
此时能够发现,合并的一定都是这些区间里的后缀。
再结合这个限制,相当于是要求只有一个区间。
贪心的考虑,因为每次合并都是合并一段后缀,那么前 \(i\) 个数形成的连通块如果不是一个区间(不满足限制),最靠前的区间的左端点一定不为 \(i\)。
于是会发现在过程中只关心最靠前的区间的右端点,那就可以考虑设计 dp 了。
设 \(f_{i, j}\) 表示前 \(i\) 个数,最靠前的区间的右端点是 \(j\)(前 \(i\) 个的相对顺序)的答案。
考虑如果加入了 \(a_{i + 1} = x\)(前 \(i + 1\) 个的相对顺序),此时 \(j\) 的变化是什么。
经过分讨容易知道有 \(j' = \begin{cases}i + 1 & 1\le x\le j\\j & j < x \le i + 1\end{cases}\)。
那么转移就很好写出了:\(f_{i + 1, i + 1}\gets f_{i, j}\times \sum\limits_{k = 1}^j a_{i + 1}^{i + 1 - k}, f_{i + 1, j}\gets f_{i, j}\times \sum\limits_{k = j + 1}^{i + 1} a_{i + 1}^{i + 1 - k}\)。
因为这都对应的是 \(a_{i + 1}^k\) 的前缀和或后缀和,所以容易优化到 \(\mathcal{O}(1)\)。
当然也可以把 \(f_{i + 1, i + 1}\) 视作“不合法”的方案数,类似总和减掉合法方案数,求出 \(f_{i + 1, j}(j\le i)\) 有 \(f_{i + 1, i + 1} = \sum\limits_{j = 1}^i f_{i, j}\sum\limits_{j = 0}^i a_{i + 1}^j - \sum\limits_{j = 1}^i f_{i + 1, j}\),这样就只需要关心 \(a_{i + 1}^k\) 的前缀和,会方便写一些。
对于 \(s_i = 1\) 的位置,要求前缀必须为一个连通块,把不合法的 \(f_{i, j}(j < i)\) 都置为 \(0\) 即可。
最后套上外层的 \(\sum\limits_{T\subseteq S}\),对于 \(s_i = 1\) 的位置,可以是为 \(1\),就只保留 \(f_{i, i}\);也可以为 \(0\),所有数都不变。
于是扩展到 \(\sum\limits_{T\subseteq S}\) 只需要加上一个 \(f_{i, i}\gets f_{i, i}\times (1 + s_i)\)。
时间复杂度 \(\mathcal{O}(n^2)\)。
#include <bits/stdc++.h>using ll = long long;constexpr ll mod = 998244353;
constexpr int maxn = 5000 + 10;int n, a[maxn];
char s[maxn];
ll f[maxn];inline void solve() {scanf("%d", &n);for (int i = 1; i <= n; i++) {scanf("%d", &a[i]);}scanf("%s", s + 1);f[1] = 1 + (s[1] - '0');for (int i = 2; i <= n; i++) {ll pw = 1, sum = 1, sumf = 0;for (int j = i - 1; j >= 1; j--) {sumf = (sumf + f[j]) % mod;f[j] = f[j] * sum % mod;pw = pw * a[i] % mod;sum = (sum + pw) % mod;}f[i] = sumf * sum % mod;for (int j = 1; j < i; j++) {f[i] = (f[i] - f[j] + mod) % mod;}f[i] = f[i] * (1 + s[i] - '0') % mod;}ll ans = 0;for (int i = 1; i <= n; i++) {ans = (ans + f[i]) % mod;}printf("%lld\n", ans);
}int main() {int t;scanf("%d", &t);while (t--) {solve();}return 0;
}