蓝桥杯 —— 网络寻路 图的构建+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个点的也都有可能通过所有三个点的进行拼接到达。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <string>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <map>
#include <sstream>
#include <queue>
#include <stack>
#include <vector>
#define INF 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
#define For(a,b) for(int i = a;i<b;i++)
#define ll long long
#define MAX_N 100010
using namespace std;

struct qxx
{
    int to;
    int next;
}edge[MAX_N];

int head[10005];
int num,ans;

void add(int v,int u)
{
    edge[num].to = u;
    edge[num].next = head[v];
    head[v] = num ++;

    edge[num].to = v;
    edge[num].next = head[u];
    head[u] = num ++;

    return ;
}


void dfs(int from,int to,int con)
{
    if(con == 3)
    {
        ans ++;
        return ;
    }
    for(int i = head[from]; i != -1; i = edge[i].next)
    {
        int u = edge[i].to;
        if(u != to)
        {
            dfs(u,from,con + 1);
        }
    }
    return ;
}

int main()
{
    int n,m;
    while (cin>>n>>m)
    {
        num = 0,ans = 0;
        mem(head,-1);
        for (int i = 0; i < m; i++)
        {
            int a,b;
            cin>>a>>b;
            add(a,b);
        }
        for (int i = 1; i <= n; i++)
            dfs(i,-1,0);
        printf("%d\n",ans);
    }
    return 0;
}