백준 1260번 : DFS와 BFS / C++

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

1260번: DFS와 BFS

간선을 저장하는 방법에 대해서 여러가지 방법이 있을거 같은데 좀더 효율적으로 간선을 저장할 수 있는 방법을 생각해야겠다.

 
#include <iostream>
#include <set>
#include <queue>
using namespace std;
 
int N, M, V;
 
set<int>* arr[1001]{};
bool DFSvisit[1001]{};
bool BFSvisit[1001]{};
 
void DFS(int index)
{
    cout << index << ' ';
    DFSvisit[index] = true;
 
    for (auto iter = arr[index]->begin(); iter != arr[index]->end(); ++iter)
    {
        if (DFSvisit[*iter] == false)
            DFS(*iter);
    }
}
 
void BFS(int index)
{
    queue<int> q;
    q.push(index);
    BFSvisit[index] = true;
    cout << index << ' ';
 
    while (!q.empty())
    {
        int f = q.front();
        for (auto iter = arr[f]->begin(); iter != arr[f]->end(); ++iter)
        {
            if (BFSvisit[*iter] == false)
            {
                q.push(*iter);
                BFSvisit[*iter] = true;
                cout << *iter << ' ';
            }
        }
        q.pop();
    }
}
 
int main(void)
{
    cin.tie(NULL);
    cout.tie(NULL);
    ios::sync_with_stdio(false);
 
    cin >> N >> M >> V;
    for (int i = 1; i <= N; ++i)
        arr[i] = new set<int>;
 
    for (int i = 0; i < M; ++i)
    {
        int x1, x2;
        cin >> x1 >> x2;
        arr[x1]->insert(x2);
        arr[x2]->insert(x1);
    }
 
    DFS(V);
    cout << '\n';
    BFS(V);
}