[BOJ] 1260번 DFS와 BFS

Rang·2021년 4월 21일
0
📝 문제 링크
📝 참고하면 좋을 DFS BFS 설명

문제 설명

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

입력

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

출력

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

예제 입력

case1 :

입력
4 5 1
1 2
1 3
1 4
2 4
3 4


case2 :

출력
1 2 4 3
1 2 3 4

입력
3 2 1
1 2
1 3

출력
1 2 3
1 2 3


  • DFS, BFS의 가장 기초가 되는 문제.
  • 동빈나 2018 알고리즘 강의에서 배운 내용을 토대로 vector를 이용하여 간선정보를 저장하는 방식을 이용해 풀었음!

코드

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
int n, m, i, j, v, vv;
bool isVisitDfs[1001], isVisitBfs[1001];
stack<int> s;
queue<int> q;
vector<int> gv[1001];
int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    cin>>n>>m>>vv;
    for(int i=0;i<m;i++){
        int a, b;
        cin>>a>>b;
        gv[a].push_back(b);
        gv[b].push_back(a);
    }
    //DFS
    s.push(vv);
    while(!s.empty()){
        v=s.top();
        s.pop();
        if(!isVisitDfs[v]){
            cout<<v<<' ';
            sort(gv[v].begin(),gv[v].end(),greater<int>());
            for(int i=0;i<gv[v].size();i++){
                if(!isVisitDfs[gv[v][i]])
                    s.push(gv[v][i]);
            }
        }
        isVisitDfs[v]=true;
    }
    cout<<'\n';
    
    //BFS
    q.push(vv);
    while(!q.empty()){
        v=q.front();
        q.pop();
        if(!isVisitBfs[v]){
            cout<<v<<' ';
            sort(gv[v].begin(),gv[v].end(),less<int>());
            for(int i=0;i<gv[v].size();i++){
                if(!isVisitBfs[gv[v][i]]){
                    q.push(gv[v][i]);
                }
            }
        }
        isVisitBfs[v]=true;
    }
    return 0;
}

0개의 댓글