[Algorithm]Baekjoon 1613번 :: 역사

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

 

#include <iostream>
#include <string>
#include <map>
#include <vector>
#include <algorithm>
#include <math.h>
#include <queue>
#include <functional>
#include <sstream>
#include <memory.h>
#include <deque>
#include <set>
#include <unordered_set>
 
using namespace std;
 
int n, k, s;
int graph[2][401][401] = { 0 };
const int maxNum = 0x3f3f3f3f;
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
 
    cin >> n >> k;
 
    memset(graph, 0x3f, sizeof(graph));
 
    for (int i = 1; i <= n; ++i)
    {
        graph[0][i][i] = graph[1][i][i] = 0;
    }
 
    for (int i = 0; i < k; ++i)
    {
        int a, b;
        cin >> a >> b;
 
        graph[0][a][b] = 1;
        graph[1][b][a] = 1;
    }
 
    for (int k = 1; k <= n; ++k)
    {
        for (int i = 1; i <= n; ++i)
        {
            if (graph[0][i][k] != maxNum)
            {
                for (int j = 1; j <= n; ++j)
                {
                    graph[0][i][j] = min(graph[0][i][j], graph[0][i][k] + graph[0][k][j]);
                }
            }
 
            if (graph[1][i][k] != maxNum)
            {
                for (int j = 1; j <= n; ++j)
                {
                    graph[1][i][j] = min(graph[1][i][j], graph[1][i][k] + graph[1][k][j]);
                }
            }
        }
    }
 
    cin >> s;
    
    vector<int> results;
    for (int i = 0; i < s; ++i)
    {
        int a, b;
        cin >> a >> b;
 
        if (graph[0][a][b] != maxNum)
        {
            results.push_back(-1);
        }
        else if (graph[1][a][b] != maxNum)
        {
            results.push_back(1);
        }
        else
        {
            results.push_back(0);
        }
    }
 
    for (int i = 0; i < results.size(); ++i)
    {
        printf("%d\n", results[i]);
    }
}
 
 

 

바로 이전에 풀었던 문제에서 조금만 바꿔주면 된다.

마찬가지로 방향 두가지에 대해 그래프를 만들고 정답조건에 맞춰서 출력해주면 된다.