BFS를 활용한, 이진트리 탐색 in C++

Purple·2021년 9월 18일
0

1. 이진트리를 형성하는 6개의 입력을 받아, root(1)부터 시작하여 BFS를 활용한 탐색을 실시하는 코드

#include <iostream>
#include <queue>
#include <vector>

using namespace std;

vector<int> graph[7];
queue<int> Q;
int ch[7];
int main() {
    freopen("input.txt", "rt", stdin);
    for(int i=1; i<=6; i++) {
        int a, b;
        cin >> a >> b;
        graph[a].push_back(b);
    }
    Q.push(1);
    ch[1] = 1;
    while(!Q.empty()) {
        int x = Q.front();
        Q.pop();
        cout << x << " ";
        for(int i = 0; i < graph[x].size(); i++) {
            if(ch[graph[x][i]] == 0) {
                ch[graph[x][i]] = 1;
                Q.push(graph[x][i]);
            }
        }
    }

    return 0;
}

BFS는 일반적으로 자료구조 queue를 사용하여 구현한다.

  • graph[a].push_back(b) : 이진트리를 방향 그래프라고 생각
  • ch[graph[x][i]] : 해당 위치를 방문하였는지 확인
    ex)
    1 2
    1 3
    2 4
    2 5
    3 6
    3 7
profile
안녕하세요.

0개의 댓글