CF 1913 D

题目描述

给定一个所有数互不相同的长度为 $N$ 的序列 $P$，你可以执行以下操作任意次：

• 选择一对 $1\le l<r\le N$，并把其中除最小值外的所有元素删除。

求最终可以得到的不同序列数量。

思路

我们考虑怎样通过删除最少的元素来删除 $i$，很明显，就是选择区间 $[l,i]$ 或 $[i,r]$，这里 $l$ 是最大的满足 $l<i\mathrm{且}{P}_l<P_i$ 的下标，而 $r$ 是最小的满足 $r>i\mathrm{且}{P}_r<P_i$ 的下标。这个可以使用单调栈求出。

我们每次都只删除这种最小区间，因为这样会使最终的方案更多。

令 $d{p}_{0/1,i}$ 表示考虑前 $i$ 个数，最后一个数选/不选的种类数。

我们有 $d{p}_{0,i}\leftarrow d{p}_{0/1,l_i}$（因为没有任何影响，随便怎么选，但如果不删除 $i$，那么 $l_i+1$ 到 $i$ 都无法删除）。以及 $d{p}_{1,i}\leftarrow d{p}_{1,j}(l_i\le j<i),d{p}_{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;
}