ZSTU OJ 3999 零基础学算法---邻接表

题目：Click here

题意：我就喜欢中文题！

分析：这个题虽然是中文题，但是还是有一点费解的。其实就是给你一棵树，是用图的形式给你的，只知道a，b之间有一条边，并不知道谁是父，谁是子。思路就是先把这个无向图用邻接表存下来，再转成有向图，最后dfs回答每一条询问。

模拟链表实现的邻接表：

 #include <bits/stdc++.h>
 using namespace std;
 const int INF = 0x3f3f3f3f;
 const int M = +;

 struct Edge {   // 邻接表的结点
     int v, next;
 }edge[M<<];

 int n, root, E, q;
 int cnt;
 bool vis[M];
 int head[M];    // 邻接表头结点

 void newedge( int u, int v )    {   // 连接一条边
     edge[E].v = v;
     edge[E].next = head[u];
     head[u] = E;
     E++;
 }

 void dfschaneg( int root )   {  // 将无向图转换成有向图
     vis[root] = true;
     for( int i=head[root]; i!=-; i=edge[i].next )
         if( !vis[edge[i].v] )   dfschaneg( edge[i].v );
         else   edge[i].v = INF;     // 将访问过的点标记INF，之后dfs时就不会再访问了
 }

 void dfs( int pos ) {   // dfs遍历子树
     cnt++;
     for( int i=head[pos]; i!=-; i=edge[i].next )
         if( edge[i].v != INF )
             dfs( edge[i].v );
 }

 int main()  {
     #ifdef ONLINE_JUDGE
     #else
         freopen( "in.txt", "r", stdin );
     #endif
     while( ~scanf("%d %d", &n, &root ) )    {
         E = ;
         memset( vis, false, sizeof(vis) );
         memset( head, -, sizeof(head) );
         for( int i=; i<n; i++ )    {
             int x, y;
             scanf("%d %d", &x, &y );
             newedge( x, y );    // 由于是无向图，所以x到y，y到x都有边
             newedge( y, x );
         }
         dfschaneg( root );
         scanf("%d", &q );
         for( int i=; i<q; i++ )    {
             int pos;
             scanf("%d", &pos );
             cnt = ;
             dfs( pos );
             printf("%d\n", cnt );
         }
     }
     return ;
 }

使用vector实现的邻接表：

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <vector>
 #define pb push_back
 using namespace std;
 const int M = 1e4+;
 const int INF = 0x3f3f3f3f;

 int n, root, q;
 vector<int> v[M];   // vector实现的邻接表，v[i]这个vector存的是与i邻接的顶点
 bool vis[M];
 int cnt;

 void dfschange( int pos )  {    // 将无向图转换成有向图
     vis[pos] = true;
     for( int i=; i<v[pos].size(); i++ )    {
         if( !vis[v[pos][i]] )    {
             dfschange( v[pos][i] );
         }
         else    {
             v[pos][i] = INF;    // 将访问过的点标记INF，之后dfs时就不会再访问了
         }
     }
     return ;
 }

 void dfs( int x )   {
     cnt++;
     for( int i=; i<v[x].size(); i++ )    {
         if( v[x][i] != INF )
             dfs( v[x][i] );
     }
     return ;
 }

 int main()  {
     while( ~scanf("%d %d", &n, &root ) )    {
         for( int i=; i<M; i++ )    {
             v[i].clear();
         }
         for( int i=; i<n-; i++ )  {
             int x, y;
             scanf("%d %d", &x, &y );
             v[x].pb(y);     // 由于是无向图，所以x到y，y到x都有边
             v[y].pb(x);
         }
         memset( vis, false, sizeof(vis) );
         dfschange( root );
         scanf("%d", &q );
         for( int i=; i<q; i++ )    {
             int x;
             scanf("%d", &x );
             cnt = ;
             dfs( x );
             printf("%d\n", cnt );
         }
     }
     return ;
 }

本人原创文章，转载请注明出处：http://www.cnblogs.com/TaoTaoCome/p/4719282.html