【蓝桥杯】网络寻路

题目

题目

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

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

如图 11 所示的网络。

1→2→3→1 是允许的。

1→2→1→2 或者 1→2→3→2 都是非法的。

分析

分析题意可得，每个信息只能被转发两次，所以这一条传递信息的路径最多由4个节点组成，可以想到使用dfs，遍历每个节点，将每个节点长度为4的路径数纪录出来，需要注意的是，每个节点不能出现多次，但是起点和终点可以是同一个节点，所以需要使用一个数组记录，是否遍历过这个点，设置好长度为4时是出口即可。

代码

#include<bits/stdc++.h>
#define pb push_back
#define ll long long
#define guanliu ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
using namespace std;
const ll maxn=1e5+7;
const ll mod=1e9+7;
const ll INF=0x3f3f3f3f;
const double pi=acos(-1);

vector<int> mp[maxn];
int vis[maxn];
int ans,b;

void dfs(int u,int d)
{
	if(d==4)
	{
		ans++;
		return ;
	}
	else{
		for(int i=0;i<mp[u].size();i++)
		{
			int v=mp[u][i];
			if(!vis[v])
			{
				vis[v]=1;
				dfs(v,d+1);
				vis[v]=0;
			}
			else if(d==3&&v==b) dfs(v,d+1); 
		}
	}
}

int main()
{
	int n,m;
	cin>>n>>m; 
	for(int i=1;i<=m;i++)
	{
		int x,y;
		cin>>x>>y;
		mp[x].pb(y);
		mp[y].pb(x);
	}
	for(int i=1;i<=n;i++)
	{
		vis[i]=1;
		b=i;
		dfs(i,1);
		vis[i]=0;
	}
	cout<<ans<<endl;
	return 0;
}