Warshall传递闭包算法

一个n个顶点有向图的传递闭包，即为一个n阶布尔矩阵T={ t _{i j} }：如果从 i 到 j 顶点间存在一条有向路径，那么矩阵第 i 行第 j 列置1 即T[i, j]=1；否则置0即T[i, j]=0。

传递闭包给出给定图各顶点之间是否存在任意长度的有向路径。

首先可以肯定的是，求一个图的传递闭包，可以用DFS和BFS来求，但是很明显，使用DFS和BFS的话，我们必须要循环|V|次，因此效率会非常低。而Warshall算法则是使用的动态规划的思想，能够有效地提高效率。

具体思想

Warshall 算法首先对有向图的顶点编号：1,2,…,k,…n

然后按顶点编号对闭包问题进行划分：

第k阶段的子问题为求只包含前k顶点的传递闭包R(k) ：只允许路径中包含编号为1到k的顶点作为中间顶点。

然后通过一系列计算得到最终的传递闭包矩阵R(n)：

其中，

每个后继矩阵 R(k) 对其前趋 R(k-1) 来说，在路径上允许增加一个顶点，因此有可能包含更多的1（增加前为1的在增加后依然为1）。

确定递推关系式

所以状态转移方程为：

如：求出下图的传递闭包

则Warshall算法的过程为：

对于R⁽⁰⁾，很显然为该图的邻接矩阵。

对于R⁽¹⁾，在R⁽⁰⁾的基础上添加了第一行第一列，然后我们可以发现[d,a]=1，因此这就说明，存在一条由d到a的路径，那么只要a能够到达的节点，d也能够到达，此时我们又发现[a,b]=1，因此应该将[d,b]置为1。

对于R⁽²⁾，在R⁽¹⁾的基础上添加了第二行第二列，由于列方向上[a,b]=1，[d,b]=1，而行方向上[b,d]=1，因此将[a,d]和[d,d]置为1。

对于R⁽³⁾，在R⁽²⁾的基础上添加了第三行第三列，由于列方向上[d,b]=1，而行方向上全为0（说明节点c无法到达其它任何节点），因此不用改变。

对于R⁽⁴⁾，在R⁽³⁾的基础上添加了第四行第四列，由于列方向上[a,d]=1，[b,d]=1，[d,d]=1，而行方向上全为1（说明节点d可以到达其它任何节点），因此第a、b、d行全部置为1，此时算法终止。

注意：由于看[ i , j ]是否可达，还要考虑[ i , k ]和[ k , j ]之间是否可达，因此我们看图的时候应该是先看列方向，再看行方向。

伪代码

由上述思想，我们可以得到如下伪代码：

algorithm Warshall(A[1..n, 1..n])
//使用Warshall计算传递闭包
//输入：邻接矩阵A[1..n, 1..n]
//输出：该有向图的传递闭包
R(0) <- A
for k <- 1 to n do
	for i <- 1 to n do
		for j <- 1 to n do
			R(k)[i,j] <- R(k-1)[i,j] OR (R(k-1)[i,k] AND R(k-1)[k,j])
return R(n)

注意：这里虽然对R加了一个上标，但是我们其实不会需要使用n个n * n的矩阵，只需要一个即可（加上标只是为了让我们自己清楚这之间的不同），因为每次迭代的过程中，第k行第k列都不会改变（从前面的例子中我们也可以发现），我们只是借助第k行第k列来修改原矩阵，因此矩阵中的值在每次迭代的过程中除了第k行第k列的值以外，其他的值要么被上次迭代的值覆盖（即不变），要么就被本次迭代产生的新值覆盖（即更新）。因此，我们只需要对一个n * n的矩阵进行n次迭代即可。

代码实现

这里我使用的是C++（以前面的例子为例）：

首先是邻接矩阵：

int A[4][4] = {
	{0, 1, 0, 0},
	{0, 0, 0, 1},
	{0, 0, 0, 0},
	{1, 0, 1, 0}
};

算法部分的实现：

void Warshall(int A[][4], int R[][4]) {
	for (int i = 0; i < 4; i++) {
		for (int j = 0; j < 4; j++) {
			R[i][j] = A[i][j];
		}
	}
	for (int k = 0; k < 4; k++) {
		for (int i = 0; i < 4; i++) {
			for (int j = 0; j < 4; j++) {
				R[i][j] = R[i][j] || (R[i][k] && R[k][j]);
			}
		}
	}
}

这里注意：由于C/C++的特性，如果为真，则默认为1，为假则默认为0，因此可以直接将布尔表达式的值赋值给一个int类型的变量。

最后在main()函数中启动：

int main() {
	int R[4][4];

	Warshall(A, R);
	for (int i = 0; i < 4; i++) {
		for (int j = 0; j < 4; j++) {
			cout << R[i][j] << " ";
		}cout << endl;
	}

	return 0;
}

运行结果如下：

可以看到，输出了我们想要的最终结果。

完整代码

#include<iostream>
using namespace std;

int A[4][4] = {
	{0, 1, 0, 0},
	{0, 0, 0, 1},
	{0, 0, 0, 0},
	{1, 0, 1, 0}
};

void Warshall(int A[][4], int R[][4]) {
	for (int i = 0; i < 4; i++) {
		for (int j = 0; j < 4; j++) {
			R[i][j] = A[i][j];
		}
	}
	for (int k = 0; k < 4; k++) {
		for (int i = 0; i < 4; i++) {
			for (int j = 0; j < 4; j++) {
				R[i][j] = R[i][j] || (R[i][k] && R[k][j]);
			}
		}
	}
}

int main() {
	int R[4][4];

	Warshall(A, R);
	for (int i = 0; i < 4; i++) {
		for (int j = 0; j < 4; j++) {
			cout << R[i][j] << " ";
		}cout << endl;
	}

	return 0;
}