본문 바로가기
백준/C++

백준 1260번 : DFS와 BFS [C++]

by 대니스 2022. 9. 5.

주소 : https://www.acmicpc.net/problem/1260

 

1260번: DFS와 BFS

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사

www.acmicpc.net

소스 코드 :

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;

int map[1001][1001];
int visit[1001];
queue<int> q;

void dfs(int node,int n)
{
	visit[node] = 1;
	cout << node << ' ';

	for (int i = 1; i <= n; i++)
	{
		if (!visit[i] && map[node][i] == 1)
		{
			dfs(i, n);
		}
	}
}

void bfs(int node, int n,int v)
{
	visit[node] = 1;
	q.push(v);
	while (!q.empty())
	{
		int x = q.front();
		cout << x << ' ';
		q.pop();

		for (int i = 1; i <= n; i++)
		{
			if (!visit[i] && map[x][i] == 1)
			{
				visit[i] = 1;
				q.push(i);
			}
		}
	}

}

int main()
{
	ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
	int N, M, V;
	cin >> N >> M >> V;

	for (int i = 0; i < M; i++)
	{
		int u, v;
		cin >> u >> v;
		map[u][v] = map[v][u] = 1;
	}

	dfs(V, N);
	memset(visit, 0, sizeof(visit));
	cout << '\n';
	bfs(V, N, V);
}

마무리 : DFS와 BFS의 문제로 기본 문제이다. 이 문제를 풀면 DFS와 BFS에 좀 더 가까워지고 두 개의 알고리즘으로 응용을 할 수 있게 될 것이다.