【ybtoj高效进阶 21288】头文件 B（线段树）（图论）

头文件 B

题目链接：ybtoj高效进阶 21288

题目大意

给你一个有向图，一开始之后 i 到 i+1 的有向边，保证这些边后面不会被操作。
然后有一些操作：加一条边，删去加的一条边，问你从一个点出发可以到多少个点。

思路

考虑往后连的边没有意义。

考虑往前连的边，那假设是 \(x\rightarrow y(x>y)\)，那说明如果你是在 \([y+1,x]\) 里面的点，你都可以逆向（向编号小的点）走至少一步。

那加边就相当于加标记，删边就相当于取消标记。
然后你考虑询问操作，那首先它后面的点是肯定能到的。
接下来就是看能不能一步一步往前走。

那就相当于要找从那个点开始向左一段连续的标记。
这个可以用线段树实现。

然后就好了。

代码

#include<cstdio>
#include<iostream>

using namespace std;

int n, m, op, x, y;

struct XDtree {//线段树
	int lzy[500001 << 2], a[500001 << 2];
	
	void up(int now) {
		a[now] = min(a[now << 1], a[now << 1 | 1]);
	}
	
	void down(int now) {
		if (!lzy[now]) return ;
		a[now << 1] += lzy[now]; a[now << 1 | 1] += lzy[now];
		lzy[now << 1] += lzy[now]; lzy[now << 1 | 1] += lzy[now];
		lzy[now] = 0;
	}
	
	void insert(int now, int l, int r, int L, int R, int va) {
		if (L <= l && r <= R) {
			lzy[now] += va; a[now] += va;
			return ;
		}
		down(now);
		int mid = (l + r) >> 1;
		if (L <= mid) insert(now << 1, l, mid, L, R, va);
		if (mid < R) insert(now << 1 | 1, mid + 1, r, L, R, va);
		up(now);
	}
	
	int query(int now, int l, int r, int R) {//找到从右边一直走没有路就停下能走多远
		if (a[now]) return min(r, R) - l + 1;
		if (l == r) return 0;
		down(now);
		int mid = (l + r) >> 1;
		if (r <= R) {
			if (a[now << 1 | 1]) return r - mid + query(now << 1, l, mid, R);
				else return query(now << 1 | 1, mid + 1, r, R);
		}
		if (mid < R) {
			int x = query(now << 1 | 1, mid + 1, r, R);
			if (x == R - mid) return x + query(now << 1, l, mid, R);
				else return x;
		}
		return query(now << 1, l, mid, R);
	}
}T;

int main() {
//	freopen("build.in", "r", stdin);
//	freopen("build.out", "w", stdout);
	
	scanf("%d %d", &n, &m);
	while (m--) {
		scanf("%d", &op);
		if (op == 1) {
			scanf("%d %d", &x, &y);
			if (x <= y) continue;
			T.insert(1, 1, n, y + 1, x, 1);
		}
		if (op == 2) {
			scanf("%d %d", &x, &y);
			if (x <= y) continue;
			T.insert(1, 1, n, y + 1, x, -1);
		}
		if (op == 3) {
			scanf("%d", &x);
			printf("%d\n", T.query(1, 1, n, x) + n - x + 1);
		}
	}
	
	return 0;
}