HDU 4707 Pet 邻接表实现

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4707

解题报告：题目大意是在无向图G中有n个点，分别从0 到n-1编号，然后在这些点之间有n-1条边，可以保证这个图是连通图，并且每条边的长度都是1，然后让你求出从编号为0的点出发，到其它的点的距离大于d的点的个数。

这题的点的个数有500000个，而内存限制为32M，很显然，开邻接矩阵不行，但这题实际上不需要记录边只需要开一位数组就可以了，但是我为了练习一下邻接矩阵的写法，特地用邻接矩阵另外写了一个代码。第一次写邻接矩阵，一开始忘了标记已经走过的点，造成dfs的时候陷入了死循环，一运行就挂了，很明显爆栈了，找了好久，才发现原来走过的点如果不标记还会再走一遍。还有一个很重要的点就是，每组数据结束之后，不要忘记清理内存，不然还是会MLE。然后这题如果不用邻接表写的话，直接定义一个一位数组也行，因为题目虽然没有说，但是实际上题目输入的数据是有顺序的，就是按照从0点开始往其它的点走的顺序来输入的，很显然，这样就方便多了。

邻接表AC代码：

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 using namespace std;
 const int maxn = ;

 struct Node
 {
     int d;
     Node *next;
 };

 typedef struct node
 {
     int num;
     Node * next;
     node()
     {
         num = ;
         next = NULL;
     }
 }LinkList;

 int length[maxn],visit[maxn];

 void push(LinkList *Link,int s,int e)
 {
     int k = Link[s].num - ;
     Node *p = Link[s].next;
     Node *q = new Node;
     q->d = e;
     q->next = NULL;
     if(p == NULL)
     Link[s].next = q;
     else
     {
         while(k--)
         p = p->next;
         p->next = q;
     }
     Link[s].num++;
 }
 void Delete(Node *p)
 {
     if(p == NULL)
     return ;
     Delete(p->next);
     delete p;
 }

 void clean(LinkList *Link,int n)
 {
     for(int i = ;i < n;++i)
     Delete(Link[i].next);
 }

 void dfs(LinkList *head,int l,int deep)
 {
     length[l] = deep;
     visit[l] = ;
     int k = head[l].num;
     Node *p = head[l].next;
     while(p != NULL)
     {
         if(visit[p->d] == )
         dfs(head,p->d,deep+);
         p = p->next;
     }
 }

 int main()
 {
     int T,n,d,a,b;
     scanf("%d",&T);
     while(T--)
     {
         LinkList Link[maxn];
         scanf("%d%d",&n,&d);
         for(int i = ;i<n;++i)
         {
             scanf("%d%d",&a,&b);
             push(Link,a,b);
             push(Link,b,a);
         }
         memset(length,,sizeof(length));
         memset(visit,,sizeof(visit));
         dfs(Link,,);
         int tot = ;
         for(int i = ;i < n;++i)
         if(length[i] > d)
         tot++;
         printf("%d\n",tot);
         clean(Link,n);
     }
     return ;
 }

一维数组AC代码：

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 using namespace std;

 const int maxn = ;
 int len[maxn];

 int main()
 {
     int T,n,d,s,e;
     scanf("%d",&T);
     while(T--)
     {
         scanf("%d %d",&n,&d);
         memset(len,,sizeof(len));
         for(int i = ;i < n;++i)
         {
             scanf("%d %d",&s,&e);
             if(len[s]) len[e] = len[s] + ;
             else if(s == ) len[e] = ;
             else ;
         }
         int tot = ;
         for(int i = ;i < n;++i)
         if(len[i] > d)
         tot++;
         printf("%d\n",tot);
     }
     return ;
 }