历届试题 网络寻路 (树上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

题解：

因为要求必须恰好被转发两次到达目的地，树上dfs就可以了。

把每个节点dfs一遍，如果能回到当前节点就dfs 3次，否则就dfs 4 次。每次对答案贡献一次。把每个节点dfs完就可以得到答案了。

加了个读入挂....1s卡过....不过多交几次.....

代码：

#include<bits/stdc++.h>using namespace std;vector<int>V[10010];int vis[10010];int n,m;int ans;int rt;//必须恰好被转发两次到达目的地void dfs(int cur,int fa){if(fa==3){ans++;return ;}for(int i=0;i<V[cur].size();i++){if(fa==2 && V[cur][i]==rt){dfs( V[cur][i], fa+1 );}if(vis[ V[cur][i] ]==1)continue;vis[V[cur][i]]=1;dfs( V[cur][i], fa+1 );vis[V[cur][i]]=0;}}int read(){int v = 0, f = 1;char c =getchar();while( c < 48 || 57 < c ){if(c=='-') f = -1;c = getchar();}while(48 <= c && c <= 57) v = v*10+c-48, c = getchar();return v*f;}int main(){    n=read();m=read();for(int i=0;i<m;i++){int x,y;x=read();y=read();V[x].push_back(y);V[y].push_back(x);}for(int i=1;i<=n;i++){memset(vis,0,sizeof vis);vis[i]=1;rt=i;dfs(i,0);}printf("%d\n",ans);    return 0;}