[C++]백준 - 1260번 문제

1260번: DFS와 BFS (acmicpc.net)

1260번: DFS와 BFS

 

1260번 : DFS와 BFS

---

그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.

 

입력

---

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

 

 

출력

---

첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.

 

 

 

---

 

생각해 볼 점

---

DFS는 스택, BFS는 큐로 풀 수 있습니다.

 

이전에 방문한 적이 있는 지를 나타내는 visited 배열을 지정하고, 각각 DFS 및 BFS의 알고리즘에 맞게 풉니다.

 

저는 간선에 대한 정보를 이차원 배열에 저장하여 활용했습니다.

 

 

이차원 배열의 활용은 다음과 같습니다.

 

코드

---

#include <iostream>
#include <stack>
#include <queue>
using namespace std;

//클래스 구현
class graph
{
private:
    int N;
    bool **matrix;
    bool *visited;
    stack<int> st;
    queue<int> q;
public:
    graph(int N, int M, int V);
    ~graph();
    void dfs();
    void bfs();
    void refresh();
};

//생성자, matrix의 동적 할당
graph::graph(int N, int M, int V)
{
    this->N = N;
    matrix = new bool*[N];
    visited = new bool[N];
    fill_n(visited, N, false);
    for(int i = 0; i < N; i++)
    {
        matrix[i] = new bool[N];
        fill_n(matrix[i], N, false);
    }   
    
    for(int i = 0; i < M; i++)
    {
        int s, d;
        cin >> s >> d;
        s--;
        d--;
        matrix[s][d] = true;    //양방향이므로 두번
        matrix[d][s] = true;
    }
    st.push(V - 1);
    q.push(V - 1);
}

//소멸자, 동적할당 해제
graph::~graph()
{
    for(int i = 0; i < N; i++) delete[] matrix[i];
    delete[] matrix;
    delete[] visited;
}

//DFS 탐색, 스택을 이용
void graph::dfs()
{
    if(st.empty()) return;
    
    int current = st.top();
    visited[current] = true;
    cout << current + 1 << " ";
    st.pop();
    for(int i = 0; i < N; i++)
    {
        if(matrix[current][i] && !visited[i])
        {
            st.push(i);
            dfs();
        }
    }
}

//BFS 탐색, 큐를 이용
void graph::bfs()
{
    if(q.empty()) return;
    
    int current = q.front();
    visited[current] = true;
    cout << current + 1 << " ";
    q.pop();
    for(int i = 0; i < N; i++)
    {
        if(matrix[current][i] && !visited[i])
        {
            q.push(i);
            visited[i] = true;
        }
    }
    bfs();
}

//visited 배열을 초기화해주는 함수
void graph::refresh()
{
    for(int i = 0; i < N; i++) visited[i] = false;
}

int main()
{
    int N, M, V;
    cin >> N >> M >> V;
    
    graph g(N, M, V);
    
    g.dfs();
    cout << "\n";
    g.refresh();
    g.bfs();
    
    return 0;
}

 

그 외

---

저는 그래프 문제를 Class로 구현하는 것을 좋아하기 때문에 굳이 클래스를 이용하였습니다. 혹시 제 글을 참고하시는 분들은 그냥 전역변수로 지정하여 푸셔도 무방할 것입니다.