问题描述
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 2
2 3
1 3
样例输出1
6
样例输入2
4 4
1 2
2 3
3 1
1 4
1 2
2 3
3 1
1 4
样例输出2
10
Thinking:
易知,
第一步:构造无向图,我这里选择了vector作为container,因为不用经过繁琐的插入删除,所以vector为首选。
第二步:在所建立的图上进行搜索。依题意,对于每一个vertex,搜索深度为3层,且在执行一次新的搜索时,不能搜索最近一次搜过的vertex。当搜索深度到达第三层,满足题意的路径数+1,之后return。
#include<iostream>
#include<vector>
using namespace std;
int counter=0;
//preNode:Previous Node ; curNode: Current Node
void dfs(int preNode,int curNode,vector<int>*edge,int step)
{
if(step==3)//当搜索深度达到三层
{
counter++;
return;
}
if(edge[curNode].size()!=0)//若当前vertex存在相连的vertex
{
vector<int>::iterator it;
for(it=edge[curNode].begin();it!=edge[curNode].end();it++)
{
if(*it!=preNode)//不能搜索最近一次搜索过的vertex
{
dfs(curNode,*it,edge,step+1);//dfs
}
}
}
}
int main()
{
int nVertex,nEdge,i,from,to;
cin>>nVertex>>nEdge;
vector<int>*edge=new vector<int>[nVertex+1];
for(i=1;i<=nEdge;i++)
{
cin>>from>>to;
edge[from].push_back(to);//构造无向图,from可以到to
edge[to].push_back(from);//to也可以到from
}
for(i=1;i<=nVertex;i++)//对所有vertex进行dfs
{
dfs(0,i,edge,0);
}
cout<<counter<<endl;
return 0;
}