PREV-13 网络寻路(dfs)

问题描述

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

6

样例输入2

4 4
1 2
2 3
3 1
1 4

样例输出2

10


分析:

题目很简单,意思就是可以从每一个结点开始走,走3步,共4个结点,最后一个结点可以是自身,dfs3次,最后一次判定一下能否到自身(因为vis = 1了),代码如下:

#include<bits/stdc++.h>

using namespace std;
const int maxn = 1e4 + 6;
vector<int > v[maxn];
int vis[maxn];
int n,m;
int cnt;
int st;

void dfs(int s,int num){
    if(num == 3){
        cnt++;
        return ;
    }
    vis[s] = 1;
    for(int i = 0;i < v[s].size();i++){
       if(num == 2 && v[s][i] == st) dfs(v[s][i],3);
       if(!vis[v[s][i]]) dfs(v[s][i],num + 1);
    }
    vis[s] = 0;
}

int main(){
    while(cin >> n >> m){
        cnt = 0;
        memset(vis,0,sizeof vis);
        for(int i = 0;i < maxn;i++) v[i].clear();
        for(int i = 1;i <= m;i++){
            int a,b;
            cin >> a >> b;
            v[a].push_back(b);
            v[b].push_back(a);
        }
        for(int i = 1;i <= n;i++){
            st = i;
            dfs(i,0);
        }
        cout << cnt << endl;
    }



    return 0;
}

猜你喜欢

转载自blog.csdn.net/guifei0/article/details/86671834