蓝桥杯 网络寻路

暴力枚举

问题描述
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

思路： 之前在bestcoder上遇到过一题，数羊的，当时由于用dfs来数4条边…导致超时，所以就对这样的题目印象深刻， 这个思路真的很有意思， 枚举每一条边，找两个端点的 除了彼此的边的数量，比如是a，b，不同的排列就有a*b种， 而且这种顺序是可以倒过来的 ，所以最后的是 ans*2 但是蓝桥杯的时间卡的不紧，所以用dfs来数也没什么问题。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <set>
using namespace std;
#define maxn 100005*2
set<pair<int,int> > vis;
int edgenum = 0;
struct node
{
    int to,next;
    node(int a,int b = -1)
    {
        to = a;  next = b;
    }
    node(){}
}edge[maxn];
int head[10005];
int add(int from,int to)
{
    edge[edgenum] = node(to,head[from]);
    head[from] = edgenum++;
}
int n,m;
int main()
{
    for(int i = 0; i < 10005; i++)
        head[i] = -1;
    scanf("%d %d",&n,&m);
    for(int i = 0; i < m ; i ++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a); 
        vis.insert(pair<int,int> (a,b));
    }
    int num = 0;

    set<pair<int,int> >::iterator it = vis.begin();
    for(; it != vis.end(); it++)
    {
        int a = it->first, b = it->second;
        int anum = 0,bnum = 0;
        for(int i = head[a]; i != -1; i = edge[i].next)
            if(edge[i].to != b) anum++;

        for(int i = head[b]; i!= -1; i= edge[i].next)
            if(edge[i].to != a) bnum++;

        num += anum*bnum;
    }
    printf("%d\n",num*2);
    return 0;
}