题目描述
给定一个所有数互不相同的长度为 \(N\) 的序列 \(P\)，你可以执行以下操作任意次：
- 选择一对 \(1\le l < r\le N\)，并把其中除最小值外的所有元素删除。
求最终可以得到的不同序列数量。
思路
我们考虑怎样通过删除最少的元素来删除 \(i\)，很明显，就是选择区间 \([l,i]\) 或 \([i,r]\)，这里 \(l\) 是最大的满足 \(l<i且 P_l<P_i\) 的下标，而 \(r\) 是最小的满足 \(r>i且 P_r<P_i\) 的下标。这个可以使用单调栈求出。
我们每次都只删除这种最小区间，因为这样会使最终的方案更多。
令 \(dp_{0/1,i}\) 表示考虑前 \(i\) 个数，最后一个数选/不选的种类数。
我们有 \(dp_{0,i}\leftarrow dp_{0/1,l_i}\)（因为没有任何影响，随便怎么选，但如果不删除 \(i\)，那么 \(l_i+1\) 到 \(i\) 都无法删除）。以及 \(dp_{1,i}\leftarrow dp_{1,j}(l_i\le j<i),dp_{0,l_i}\)，因为删掉 \(i\) 就意味着删掉 \(l_i+1\) 到 \(i\)，而 \(l_i\) 是既可以删也可以不删。这个可以使用前缀和优化。
时空复杂度均为 \(O(N)\)。
代码
#include<bits/stdc++.h>
using namespace std;const int MAXN = 300001, MOD = 998244353;int t, n, a[MAXN], stk[MAXN], top, dp[2][MAXN], sum[MAXN];void Solve() {cin >> n;for(int i = 1; i <= n; ++i) {cin >> a[i];}top = 0;dp[1][0] = sum[0] = 1;for(int i = 1; i <= n; ++i) {for(; top && a[stk[top]] >= a[i]; --top) {}int j = stk[top];dp[0][i] = (j ? (dp[1][j] + dp[0][j]) % MOD : 0);dp[1][i] = (0ll + dp[0][j] + sum[i - 1] - (j ? sum[j - 1] : 0) + MOD) % MOD;sum[i] = (sum[i - 1] + dp[1][i]) % MOD;stk[++top] = i;}cout << (dp[0][n] + dp[1][n]) % MOD << "\n";
}int main() {ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);for(cin >> t; t--; Solve()) {}return 0;
}
