题目描述:
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)。
输入数据保证任意两点最多只有一条边连接,并且没有自己连自己的边,即不存在重边和自环。
输出:
输出一个整数,表示满足要求的路径条数。
样例:
3 3
1 2
2 3
1 3
结果:
6
思路:用二维数组每次深搜的for循环长度都是n,但用vector每次深搜的长度就是子节点的长度,搜索次数大大减少
代码:
#include<bits/stdc++.h>
#define MAX 10005
using namespace std;
vector<int> net[MAX];
int book[MAX],cnt=0;
void dfs(int start,int next,int sum)
{
for(int i=0;i<net[next].size();i++)
{
if(sum==2&&(net[next][i]==start||book[net[next][i]]==0))
{
cnt++;
continue;
}
if(book[net[next][i]]==0)
{
book[net[next][i]]=1;
dfs(start,net[next][i],sum+1);
book[net[next][i]]=0;
}
}
return ;
}
int main()
{
int n,m,i,u,v;
scanf("%d%d",&n,&m);
for(i=0;i<m;i++)
{
scanf("%d%d",&u,&v);
net[u].push_back(v);
net[v].push_back(u);
}
for(i=1;i<=n;i++)
{
book[i]=1;
dfs(i,i,0);
book[i]=0;
}
printf("%d\n",cnt);
return 0;
}