网络寻路 DFS

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

时间限制: 1Sec 内存限制: 128MB 提交: 180 解决: 77

题目描述

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

思路：

利用DFS，枚举每一个起点可以走的路径，由于路径经过的点最多是4个，所以复杂度很小。

两种情况：起点和终点相同、起点和终点不同的情况，那么我们需要讨论这两种情况就好了。

代码：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=10000+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
vector<int>edge[maxn];
int ans;
bool vis[maxn];
void dfs(int root,int len,int now)
{
    if(len==4)
    {
        ans++;
        return ;
    }
    if(len>4)
        return;
    int s=edge[now].size();
    for(int i=0; i<s; i++)
    {
        int v=edge[now][i];
        if((v==root&&len==3))
        {
            dfs(root,len+1,v);//解决重复走的问题，起点和终点相同的情况。
        }
        if(!vis[v])
        {
            vis[v]=1;
            dfs(root,len+1,v);//解决重复走的问题
            vis[v]=0;
        }
    }
}
int main()
{
    int n,m;
    cin>>n>>m;
    rep(i,1,m)
    {
        int u,v;
        cin>>u>>v;
        edge[u].push_back(v);
        edge[v].push_back(u);
    }
    rep(i,1,n)
    {
        //memset(vis,0,sizeof(vis));
        vis[i]=1;
        dfs(i,1,i);
        vis[i]=0;
    }
    cout<<ans<<endl;

}