백준 2606 - 바이러스 소스 코드

 www.acmicpc.net/problem/2606

2606번: 바이러스

 

[ 백준 2606(바이러스) 스택을 1차원 배열로 구현해서 풀이 ]

#include <cstdio>
int main()
{
	int i,n,m,a,b,ans=0;
	int stack[101],top=0;
	bool check[101]={false, };
	bool d[101][101] = {false, };
	
	scanf("%d", &n);
	scanf("%d", &m);
	for(i=0;i<m;i++)
	{
		scanf("%d %d", &a, &b);
		d[a][b] = true;
		d[b][a] = true;
	}
	check[1]=true;
	stack[top]=1;
	top++;
	do{
		a=stack[top-1];
		top--;
		if(top < 0)
		{
			break;
		}
		for(i=1;i<=n;i++)
		{
			if(check[i] == false && d[a][i] )
			{
				stack[top]=i;
				top++;
				check[i]=true;
			}
		}
	}while(1);
	
	for(i=2;i<=n;i++)
	{
		if(check[i])
			ans++;
	}
	printf("%d", ans);
	
	return 0;
}