[算法机试模拟题]1007.小黄车

【问题描述】

随着共享经济的兴起，大学城如今到处可见ofo小黄车. 小左现在打算每天都骑小黄车从宿舍去实验室. 假设大学城的地图可以简化为一个有向图，图中有N个地点（节点），用0到N-1进行编号，有些地点之间存在有向的道路（有向边）. 小左的宿舍所在地点编号为0，实验室所在地点编号为N-1. 小左希望为连续的M天规划线路，使得每天从宿舍到实验室，都至少会经过一条之前没有走过的道路（有向边）. 小左想知道M的最大值，你能帮助他么？

 

请实现下面Solution类中的countPath函数，完成上述功能.

参数G: N*N（2 <= N <= 50）邻接矩阵，如果地点i到地点j之间有道路，则G[i][j] = 1；否则G[i][j] = 0. G[i][i]的值总是为0.

返回值：M的最大值. 如果不存在满足要求的路径则返回0.

 

class Solution {

public:

int countPaths(vector<vector<int>> G) {

}          

};

 

例1：

G = {{0, 1}, {1, 0}}，返回值为2，因为第1天：0 --> 1，第2天 0 --> 1 --> 0 --> 1. 虽然小左第2天兜了一下圈，但他确实走了一条第1天没有走过的边1 --> 0.

 

例2：

G = {{0, 1, 1}, {1, 0, 1}, {1, 0, 0}}，返回值为4.

第1天：0 --> 2

第2天：0 --> 2 --> 0 --> 2

第3天：0 --> 1 --> 2

第4天：0 --> 1 --> 0 --> 1 --> 2

 

例3：

G = {{0, 1, 0}, {1, 0, 0}, {0, 0, 0}}，返回值为0.

【解题分析】

有n个顶点的一个图，求从节点0到节点n-1的路径最大数，其中每条路径必须包含一条以前没有走过的边。

我们可以这样考虑，先删除所有不能从节点0到达的节点以及该节点所关联的边，因为节点0不能到达，对于我们的结果是没有任何影响的；其次，删除所有不能到达节点n-1节点的节点以及关联的边，这对于我们的结果也没有影响。删除这些节点和边形成了一个新图，其大小为m个节点，k条边。

最后，我们每经过一个节点必须“消耗”两条边，也即是一条入边，一条出边，但我们的起点和终点边是不需要的，也即是m-2个节点；于是我们能得到(k-(m-2))条路径。

举例：

1、0.....v.....n-1，v是原来没经过的节点，我们可以说是一个节点抵消了一条边，得到了一条路径；

2、0.....v...u.....n-1，v和u是原来没经过的节点，我们可以说是两个节点抵消了两条边，得到了一条路径；

3、0.....v...u....n-1，v和u之间有原来没走过的边，一条边得到了一条路径。

【源代码】

class Solution {
public:
	int countPaths(vector<vector<int> > G) {
		int result = 0;
		int m = G.size();
		int n = G[0].size();
		if (m < 2 || n < 2 || n > 50 || m > 50)	return 0;
		// 该点是否被遍历
		vector<bool> vis(m,0);
		
		// 该点是否能到终点 
		vector<bool> to_des(n,0);
		queue<int> visited;
		
		// 删除起点不能到达的点 
		visited.push(0);
		vis[0] = true;
		while(!visited.empty()) {
			int t = visited.front();
			visited.pop();
			for (int i = 0; i < m; i++) {
				if(G[t][i] == 1 && !vis[i] && t != i) {
					vis[i] = true;
					visited.push(i);
				}
			}
		}
		if (!vis[n-1])	return 0;
		
		// 删除不能到达终点的点 
		visited.push(n-1);
		to_des[n-1] = true;
		while(!visited.empty()) {
			int t = visited.front();
			visited.pop();
			for (int i = 0; i < m; i++) {
				if (G[i][t] == 1 && !to_des[i] && i != t) {
					to_des[i] = true;
					visited.push(i);
				}
			}
		}
		
		//计算剩余的边 
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				if (G[i][j] == 1 && vis[i] && vis[j] && to_des[i] && to_des[j]) {
					result++;
				}
			}
		}
		for (int i = 0; i < n; i++) {
			if (to_des[i] && vis[i]) {
				result--;
			}
		}
		return result+2;
	}
};