网络寻路（多起点搜索）蓝桥杯省赛

[蓝桥杯][历届试题]网络寻路

时间限制: 1Sec 内存限制: 128MB 提交: 458 解决: 187

题目描述

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)。
输入数据保证任意两点最多只有一条边连接，并且没有自己连自己的边，即不存在重边和自环。

输出

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

样例输入

4 4
1 2
2 3
3 1
1 4

样例输出

10

确定一个起点搜索一次，终点搜到第三个点就可以了，第四个点在终点处找第三个点相邻的点，下面两种情况都符合要求
1.第四个点没有标记过
2.第四个点是起点

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
int n,m,ans;
bool vis[10005];
vector<int> ve[10005];
void dfs(int now,int step,int start){
	if(step==3){//到了第三个点 
		for(int i=0;i<ve[now].size();i++){//找出和第三个点相邻的点 
			int q=ve[now][i];//第四个点
			if(!vis[q]||q==start){//第四个点没有去过，或者第四个点是起点 
				ans++;
			} 
		}
		return;
	}
	for(int i=0;i<ve[now].size();i++){//直接找相邻的点搜索
		if(!vis[ve[now][i]]){
			vis[ve[now][i]]=true;
			dfs(ve[now][i],step+1,start);
			vis[ve[now][i]]=false;//回溯
		}
	}
}
int main(){
	cin>>n>>m;
	for(int i=0;i<m;i++){
		int u,v;
		cin>>u>>v;
		ve[u].push_back(v);//建立邻接表 
		ve[v].push_back(u);
	}
	for(int i=1;i<=n;i++){//确定一个搜索起点
		vis[i]=true;//起点标记 
		dfs(i,1,i);
		vis[i]=false;
	}
	cout<<ans<<endl;
	return 0;
}