错过的比赛——ARC215： D 题题解

1.前言

十几天没动了，好不容易找到时间去锻炼一下，结果就错过了比赛，而且自己应该能加大把 rating 的。 /ts

2.题解

---

这道题很像这道题，所以证明就不写得特别具体了。

---

状态定义：

假设现在枚举到了 $i$, 令 $dp[j]$ 表示子序列以 $a[j]$ 结尾的方案数。

---

状态转移：

1. $\forall j\in [1,i),a[j]\ne a[i]$

$dp[i]=({\sum }_{k=1}^{i-1}dp[k])+1$

2. $\exists j\in [1,i),a[j]=a[i]$

令 $p=\max j(j\in [1,i),a[j]==a[i])$

则 $dp[i]={\sum }_p^{i-1}dp[k](\forall q<k,a[q]\ne a[k])$

具体证明类似这篇题解，这里只写大致思路。

第一种情况：易证。

第二种情况：

将 $a[i]$ 接在小于 $p$ 的部分，则一定可以用 $a[p]$ 来代替。
将 $a[i]$ 接在等于 $p$ 的部分，则除了 $dp[p]$ 包含的序列一定不会重复，所以把 $dp[p]$ 赋为 0。
将 $a[i]$ 接在大于 $p$ 的部分，则一定不会和任何序列重复。

---

实现：

发现式子是一个 $O(n^2)$ 的，我们考虑用数据结构优化。

如果发现 $\exists j<i,a[j]==a[i]$，就将 $res-=dp[j]，dp[j]=0$，正确性显然，就不证明了。

---

参考代码（注释掉的是$O(n^2)$暴力）

#include <set>
#include <map>
#include <cmath>
#include <stack>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define PII pair <int, int>
#define LL long long
#define ULL unsigned long long

template <typename T> void read (T &x) { x = 0; T f = 1;char tem = getchar ();while (tem < '0' || tem > '9') {if (tem == '-') f = -1;tem = getchar ();}while (tem >= '0' && tem <= '9') {x = (x << 1) + (x << 3) + tem - '0';tem = getchar ();}x *= f; return; }
template <typename T> void write (T x) { if (x < 0) {x = -x;putchar ('-');}if (x > 9) write (x / 10);putchar (x % 10 + '0'); }
template <typename T> void print (T x, char ch) { write (x); putchar (ch); }
template <typename T> T Max (T x, T y) { return x > y ? x : y; }
template <typename T> T Min (T x, T y) { return x < y ? x : y; }
template <typename T> T Abs (T x) { return x > 0 ? x : -x; }

const int Maxn = 2 * 1e5;
const LL Mod = 998244353;

int n, a[Maxn + 5];
int idx[Maxn + 5];
LL dp[Maxn + 5];

LL BIT[Maxn + 5];
int lowbit (int x) { return x & -x; }
void Update (int Index, LL x) {
	for (int i = Index; i <= Maxn; i += lowbit (i))
		BIT[i] = (BIT[i] + x + Mod) % Mod;
}
LL Query (int Index) {
	LL res = 0;
	for (int i = Index; i >= 1; i -= lowbit (i))
		res = (res + BIT[i] + Mod) % Mod;
	return res;
}
LL Calc (int l, int r) {
	return (Query (r) - Query (l - 1) + Mod) % Mod;
}

int main () {
	read (n);
	for (int i = 1; i <= n; i++) {
		read (a[i]);
	}
	
	LL res = 0;
	for (int i = 1; i <= n; i++) {
		if (idx[a[i]] == 0) {
			/*
			dp[i] = pre[i - 1] + 1;
			*/
			dp[i] = Query (i - 1) + 1;
		}
		else {
			/*
			dp[i] = (pre[i - 1] - pre[idx[a[i]] - 1] + Mod) % Mod;
			res = (res - dp[idx[a[i]]] + Mod) % Mod;
			for (int j = idx[a[i]]; j < i; j++) {
				pre[j] -= dp[idx[a[i]]];
				pre[j] = (pre[j] + Mod) % Mod;
			}
			*/
			dp[i] = Calc (idx[a[i]], i - 1);
			res = (res - dp[idx[a[i]]] + Mod) % Mod; 
			Update (idx[a[i]], -dp[idx[a[i]]]);
		}
		/*
		idx[a[i]] = i;
		pre[i] = pre[i - 1] + dp[i];
		res += dp[i];
		res %= Mod; dp[i] %= Mod; pre[i] %= Mod;
		*/
		idx[a[i]] = i;
		Update (i, dp[i]);
		res += dp[i];
		res %= Mod;
	}
	write (res);
	return 0;
}