본문 바로가기
algorithm/dfs,bfs

[백준/1260] DFS와 BFS

by blogsy 2020. 4. 4.

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

 

1260번: DFS와 BFS

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.

www.acmicpc.net

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
#include <cstring>
using namespace std;

vector<int> a[1001];
bool check[1001];


void dfs(int start) {
	check[start] = true;
	cout << start << ' ';
	for (int i = 0; i < a[start].size(); i++) {
		if (!check[a[start][i]]) {
			dfs(a[start][i]);
		}
	}
}

void bfs(int start) {
	memset(check, false, sizeof(check));
	queue<int> q;
	q.push(start);
	check[start] = true;
	while (!q.empty()) {
		int node = q.front();
		cout << node << ' ';
		q.pop();
		for (int i = 0; i < a[node].size(); i++) {
			if (!check[a[node][i]]) {
				q.push(a[node][i]);
				check[a[node][i]] = true;
			}
		}
	}
}

int main() {
	int n, m, start;
	cin >> n >> m >> start;

	for (int i = 0; i < m; i++) {
		int u, v;
		cin >> u >> v;
		a[u].push_back(v);
		a[v].push_back(u);
	}
	for (int i = 1; i <= n; i++) {
		sort(a[i].begin(), a[i].end());
	}
	dfs(start);
	cout << '\n';
	bfs(start);

	return 0;
}

'algorithm > dfs,bfs' 카테고리의 다른 글

[프로그래머스] 단어 변환  (0) 2020.08.25
[백준/2667] 단지번호붙이기  (0) 2020.04.06
[백준/2606] 바이러스  (0) 2020.04.04

댓글