programmers 여행경로

programmers : https://programmers.co.kr/learn/challenges?selected_part_id=12421

 

programmers DFS/BFS level3 여행경로 문제입니다.

 

처음엔 이걸 어떤식으로 해야 효율적인가 생각해봤는데 답이 안나왔습니다.

문제를 다시 읽어보니 범위는 2~1만 이라는 힌트를 볼 수 있었습니다.

 

1만정도의 테스트케이스는 N^2로도 충분히 해결할 수 있습니다.

 

추가로 알아야 할 것은

경로를 출력 ==> DFS 를 이용해야함을 알 수 있습니다. BFS를 이용하기에는 추가 메모리가 필요하니..(?)

알파벳 순서가 앞선 출력 ==> 정렬을 해야합니다.

 

#include <string>
#include <vector>
#include <algorithm>
using namespace std;
 
vector<string> solve;
vector<bool> visit;
 
bool dfs(string now, vector<vector<string>>& tickets, int cnt) {
    // 같은 사이즈이면 답이됨
    if (cnt == tickets.size()) {
        solve.push_back(now);
        return true;
    }
 
    // 일반적인 n^2 형태
    for (int i = 0; i < tickets.size(); i++) {
        if (tickets[i][0].compare(now) == 0 && !visit[i]) {
            visit[i] = 1;
            if (dfs(tickets[i][1], tickets, cnt + 1)) {
                solve.push_back(tickets[i][0]);
                return true;
            }
            visit[i] = 0;
        }
    }
    return false;
}
 
vector<string> solution(vector<vector<string>> tickets) {
    vector<string> answer;
 
    sort(tickets.begin(), tickets.end());
 
    // 방문을 확인하기 위한 배열
    for (int i = 0; i < tickets.size(); i++) visit.push_back(0);
    
    dfs("ICN", tickets, 0);
    
    // 역으로 만들어준다
    for (int i = solve.size() - 1; i >= 0; i--) answer.push_back(solve[i]);
    
    return answer;
}