2606 바이러스

https://www.acmicpc.net/problem/2606

메모리풀로 그래프를 만들고 1번에 연결된 컴퓨터의 개수를 세준다.

#include <stdio.h>
#define N 100
 
struct NODE
{
    int node;
    struct NODE *next;
};
 
NODE HEAD[N+1],POOL[N*N];
int n,m,u,v,cnt,ans,visit[N+1];
 
void make(int u,int v)
{
    NODE *node=&POOL[cnt++];
    node->node=v;
    node->next=HEAD[u].next;
    HEAD[u].next=node;
}
 
void bfs()
{
    int queue[N*N],front=0,back=0;
    visit[1]=1;
    queue[back++]=1;
    while(front<back)
    {
        int now=queue[front++];
        for(NODE *ptr=HEAD[now].next;ptr;ptr=ptr->next)
        {
            int node=ptr->node;
            if(visit[node]) continue;
            visit[node]=1;
            queue[back++]=node;
            ans++;
        }
    }
}
 
int main()
{
    scanf("%d %d",&n,&m);
    while(m--)
    {
        scanf("%d %d",&u,&v);
        make(u,v);
        make(v,u);
    }
    bfs();
    printf("%d\n",ans);
    return 0;
}