2021.08.20 CF1552F Telepanting

$$\text{CF1552F Telepanting}$$

$\text{[Problem]}$

有一只初始在位置 $0$ 的蚂蚁，它每秒钟会向右走 $1$ 一个单位长度。

在地图上有 $n$ 个虫洞，第 $i$ 个虫洞在位置 $x_i$，如果它是活动的，它就会把蚂蚁送回到位置 $y_i(y_i<x_i)$。

初始时每个虫洞有一个参数 $s_i$，$s_i$ 是 $0$ 表示虫洞是初始静止的，而 $1$ 表示虫洞初始是活动的。

当蚂蚁踩到虫洞 $i$ 时：

• 如果这个虫洞是静止的，那么此虫洞会变成活动的，但是蚂蚁的位置不改变。
• 如果这个虫洞是活动，那么此虫洞将变成静止，且会把蚂蚁送回 $y_i$ 位置，而接下来蚂蚁将保持向右每秒 $1$ 个单位的速度前进（即下一秒它会到位置 $y_i+1$）。

现在我们想知道，这只蚂蚁需要多少时间才能到位置 $(x_n+1)$ 呢？答案对 $998244353$ 取模。

1 ≤ n ≤ 2 × 1 0 5 , ∀ i ∈ [ 1 , n ] , 1 ≤ y i < x i ≤ 1 × 1 0 9 , s i ∈ { 0 , 1 } 1 \leq n \leq 2 \times 10^{5},\quad \forall i \in [1,n], \quad 1 \leq y_{i}<x_{i} \leq 1 \times 10^{9},s_{i} \in \{0,1 \} 1≤n≤2×105,∀i∈[1,n],1≤yi​<xi​≤1×109,si​∈{0,1}。

保证 $x_{1\cdots n}$ 严格递增。

Translated by myself.

$\text{[Solution]}$

首先，如果这只蚂蚁到达了虫洞 $i$，那么代表它经过虫洞 $1\cdots (i-1)$ 时，这些虫洞一定是静止的，同时，因为它经过了这些虫洞，所以现在 这些虫洞一定是活动的。

我们记 $f_i$ 表示走到虫洞 $i$ 并且被送回位置 $y_i$ 后还需多少时间回到 $x_i$。显然对于所以 $x_j>y_i$ 的虫洞 $j(j<i)$ 都会有 $f_j$ 的贡献（因为它们一定是活动的），加上从 $y_i$ 走到 $x_i$ 的时间，我们有：

$$f_i=x_i-y_i+\sum _{x_j>y_i,j<i}{f}_j$$

很显然，因为题目说了 $x_{1\cdots n}$ 是严格递增的，所以合法的 $j$ 是集中在一个区间内的。所以我们可以用前缀和快速地维护 $\sum _{x_j>y_i,j<i}{f}_j$ 这一串的值。

最后的答案就是：

$$\sum _{i=1}^n{f}_i[s_i=1]+x_n+1$$

前一半表示因为虫洞耽搁的时间，后一半（$x_n+1$）表示没有虫洞直接走到 $(x_n+1)$ 也必须要的时间。

总的时间复杂度：$\mathcal{O}(n)$。总的空间复杂度：$\mathcal{O}(n)$。

$\text{[Code]}$

const int N=2e5+100;
const int mod=998244353;
long long f[N],pre[N],ans;
bool s[N];int x[N],y[N],n;
int main(){
	n=read();ans=0ll;
	for(int i=1;i<=n;i++){
		x[i]=read();y[i]=read();s[i]=read();
		int j=lower_bound(x+1,x+i+1,y[i])-x;
		f[i]=(x[i]-y[i]+pre[i-1]+mod-pre[j-1])%mod;
		ans+=s[i]?f[i]:0;pre[i]=pre[i-1]+f[i];
	}
	printf("%lld",(ans+x[n]+1)%mod);
	return 0;
}