【蓝桥杯】网络寻路

问题描述

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

import java.util.Scanner;

public class Main {
    private static int count = 0;
    private static int n;
    private static int[] book;
    private static int[] a;
    private static int[][] map;

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        int m = sc.nextInt();
        map = new int[n + 1][n + 1];
        for (int i = 0; i < m; i++) {
            int p1 = sc.nextInt();
            int p2 = sc.nextInt();
            map[p1][p2] = 1;
            map[p2][p1] = 1;
        }
        book = new int[n + 1];
        a = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            book = new int[n + 1];// 每次换起点的时候，重新清空book数组
            dfs(i, 0, i);
        }
        // dfs(2, 0, 2);
        System.out.println(count);
        // sop(a);
    }

    private static void sop(int[] arr)// 用来打印每条符合的路径
    {
        // TODO Auto-generated method stub
        for (int i = 0; i <= n; i++) {
            System.out.print(arr[i] + "  ");
        }
        System.out.println();
    }

    private static void dfs(int nod, int step, int s) {
        // TODO Auto-generated method stub
        book[nod] = 1;
        if (step >= 3) {
            count++;
            // sop(a);用来打印每条符合的路径
            return;
        }
        for (int i = 1; i <= n; i++) {
            if (book[i] == 0 && map[nod][i] == 1 || (i == s && step == 2 && map[nod][i] == 1)) // 在符合未访问节点的情况下
            // 或者
            // 在访问够了2个节点
            // 才能继续访问起点，即step== 2 的时候才可以再次访问起点
            {
                a[step] = i;// 用来记录每一条符合要求的路径
                dfs(i, step + 1, s);
                if (i != s)// 当i == s 时不用重置 0 因为总是从s为起点，s一直被标记为访问过了
                {
                    book[i] = 0;
                }
            }
        }
    }

}