蓝桥杯 网络寻路 JAVA 非满分做法

问题描述

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

思路

深搜+记步长，但是最后会超时。

package LanQiao;

import java.util.*;
public class 网络寻路 {

    static int n,m;
    static int u,v;
    static int[][] map;
    static boolean vis[]= {false};
    static int res=0;
    
    public static void dfs(int start,int end,int count) {
        if(count==3) {
            if(start==end) {res++;}
            return;
        }
        if(start==end && count!=0) return;
        for(int i=1;i<=n;i++) {
            if(map[start][i]==1 && (vis[i]==false || i==end)) {
                vis[i]=true;
                dfs(i,end,count+1);
                vis[i]=false;
            }
        }
    }
    
    public static void getResult() {
        for(int i=1;i<=n;i++) {
            for(int j=1;j<=n;j++) {
                vis[i]=true;
                dfs(i,j,0);
                vis[i]=false;
            }
        }
        System.out.println(res);
    }
    
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner scan=new Scanner(System.in);
        n=scan.nextInt();
        m=scan.nextInt();
        map=new int[n+1][n+1];
        vis=new boolean[n+1];
        for(int i=0;i<m;i++) {
            u=scan.nextInt();
            v=scan.nextInt();
            map[u][v]=1;
            map[v][u]=1;
        }
        getResult();
    }

}