牛客挑战赛47 C - 条件 (bitset优化Floyd)

题目链接: 条件

大致题意

QAQ

解题思路

Floyd

读完题目, 很明显是多源最短路的考察. 而本题的数据范围为1E3, 我们是跑不下O(n³)的Floyd算法的.
考虑到优化: 用bitset优化, 复杂度为O(n³ / 32).

我们用g[N][N]数组表示两点是否可达.

用k, i, j表示Floyd的三个循环变量, 通常的传递过程为g[i][j] |= g[i][k] and g[k][j].
而对于bitset优化的floyd, 我们取消掉第三层j循环, 如果g[i][k], 则g[i] |= g[k].

为什么我们可以直接这样做呢? 我们如果用bitset去优化, 则定义: bitset<N> g[N]

其中 g[i] 用二进制的形式表示i这个点都能到那些点, 若能达, 则对应位上为1.
而g[i] |= g[k]的含义是, 对于某个点j而言, 若i能达, 或者k能达, 则g[i][j]可达.

如果g[i][k]成立, 则i和k的可达点是相通的, 因此可以写成g[i] |= g[k].

关于本题的数据处理方式: 我们用g1数组表示一定可达, g2数组表示可能可达.

由于给出的m2条边是不可达边, 不可达不具有传递性, 我们要进行转换成可达情况.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 1E3 + 10;
bitset<N> g1[N], g2[N];
bitset<N> vis[N];
int main()
{
	int n, m1, m2, q; cin >> n >> m1 >> m2 >> q;
	rep(i, n) g1[i][i] = 1; //这步不要忘了, 自身都是可达的.

	rep(i, m1) {
		int x, y; scanf("%d %d", &x, &y);
		g1[x][y] = 1;
	}
	rep(i, m2) {
		int x, y; scanf("%d %d", &x, &y);
		vis[x][y] = 1;
	}

	rep(i, n) rep(j, n) if (!vis[i][j]) g2[i][j] = 1; //不是不可达, 就可能可达

	rep(k, n) rep(i, n) {
		if (g1[i][k]) g1[i] |= g1[k];
		if (g2[i][k]) g2[i] |= g2[k];
	}

	while (q--) {
		int x, y; scanf("%d %d", &x, &y);
		printf("%s %s\n", g1[x][y] ? "Yes" : "No", g2[x][y] ? "Yes" : "No");
	}
	return 0;
}

END