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 = 100005;

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

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

int length[maxn],visit[maxn];

void push(LinkList *Link,int s,int e)
{
    int k = Link[s].num - 1;
    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 = 0;i < n;++i)
    Delete(Link[i].next);
}
        
void dfs(LinkList *head,int l,int deep)
{
    length[l] = deep;
    visit[l] = 1;
    int k = head[l].num;
    Node *p = head[l].next;
    while(p != NULL)
    {
        if(visit[p->d] == 0)
        dfs(head,p->d,deep+1);
        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 = 1;i<n;++i)
        {
            scanf("%d%d",&a,&b);
            push(Link,a,b);
            push(Link,b,a);
        }
        memset(length,0,sizeof(length));
        memset(visit,0,sizeof(visit));
        dfs(Link,0,0);
        int tot = 0;
        for(int i = 1;i < n;++i)
        if(length[i] > d)
        tot++;
        printf("%d
",tot);
        clean(Link,n);
    }
    return 0;
}

一维数组AC代码：

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

const int maxn = 100005;
int len[maxn];

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