蓝桥杯 PREV-13 网络寻路

 历届试题 网络寻路  

时间限制：1.0s   内存限制：256.0MB

      

问题描述

X 国的一个网络使用若干条线路连接若干个节点。节点间的通信是双向的。某重要数据包，为了安全起见，必须恰好被转发两次到达目的地。该包可能在任意一个节点产生，我们需要知道该网络中一共有多少种不同的转发路径。

源地址和目标地址可以相同，但中间节点必须不同。

如下图所示的网络。

1 -> 2 -> 3 -> 1 是允许的

1 -> 2 -> 1 -> 2 或者 1 -> 2 -> 3 -> 2 都是非法的。

输入格式

输入数据的第一行为两个整数N M，分别表示节点个数和连接线路的条数(1<=N<=10000; 0<=M<=100000)。

接下去有M行，每行为两个整数 u 和 v，表示节点u 和 v 联通(1<=u,v<=N , u!=v)。

输入数据保证任意两点最多只有一条边连接，并且没有自己连自己的边，即不存在重边和自环。

输出格式

输出一个整数，表示满足要求的路径条数。

样例输入1

3 3
1 2
2 3
1 3

样例输出1

6

样例输入2

4 4
1 2
2 3
3 1
1 4

样例输出2

10

分析：由于路径长度仅为4，可以暴力求解所有路径，然后排除不合要求的路径。一般做法是DFS遍历求取符合要求的路径。有一种巧妙一点的变化：当路径长度统计到3时，由于结束点可与起始点相同，结束的真正不可取的只有第二个路径点，其余所有第三的路径点的邻接节点都可以取，因此此时路径数可直接加第三个路径点的邻接节点数 - 1。这种做法可以节省大量时间。

代码：

#pragma warning(disable:4996)
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> visit;
vector<vector<int> > G;
int N, M;
long long ans = 0;
void dfs(int start, int len, int root) {
	if (len != 1 && start == root) {
		return;
	}
	if (len > 3) {
		return;
	}
	if (len == 3) {
		ans += G[start].size() - 1;
		return;
	}
	for (int i = 0; i < G[start].size(); i++) {
		int v = G[start][i];
		if (visit[v] == 0) {
			visit[v] = 1;
			dfs(v, len + 1, root);
			visit[v] = 0;
		}
	}
}
void dfs() {
	for (int i = 1; i <= N; i++) {
		dfs(i, 1, i);
	}
}
int main() {
	cin >> N >> M;
	G.resize(N + 1);
	visit.resize(N + 1);
	for (int i = 1; i <= M; i++) {
		int u, v;
		scanf("%d %d", &u, &v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs();
	cout << ans << endl;
	return 0;
}