Judge - 1231 : 양궁 선수 순위 예측 C++

출처 : https://judge.koreatech.ac.kr/problem.php?id=1231

 

이기는 경우와 지는 경우를 나눠서 그래프 두개를 만들어서 풀었다.

정답을 저장하는 벡터에 초기값으로 1,n 을 pair로 저장하고 ( 처음에는 1~n등 전부 가능한 상태)

이기는 그래프를 탐색해서 k개의 탐색을 했다면 내 밑에 k명이 있기 때문에 1~n-k등이 가능하다.

지는 그래프를 탐색해서 p개의 탐색을 했다면 내 위에 p명이 있기 때문에 1+p ~ n-k 등이 가능하다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, m;
int cnt = -1;

bool isCycle = false;

vector<vector<int>> win;
vector<vector<int>> lose;
vector<bool> visited;
vector<pair<int, int>> ans;

void Refresh() {
    cnt = -1;
    for (int i = 1; i <= n; i++) {
        visited[i] = false;
    }
}

void DfsWin(int curr, int start) {
    if (visited[curr] && curr == start) {
        isCycle = true;
        return;
    }
    if (visited[curr]) return;
    visited[curr] = true;
    cnt++;
    for (auto next : win[curr]) {
        DfsWin(next, start);
    }
}

void DfsLose(int curr, int start) {
    if (visited[curr] && curr == start) {
        isCycle = true;
        return;
    }
    if (visited[curr]) return;
    visited[curr] = true;
    cnt++;
    for (auto next : lose[curr]) {
        DfsLose(next, start);
    }
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> m;
    win = vector<vector<int>>(n + 1, vector<int>());
    lose = vector<vector<int>>(n + 1, vector<int>());
    ans = vector<pair<int, int>>(n + 1, { 1,n });
    visited = vector<bool>(n + 1, false);

    for (int i = 0; i < m; i++) {
        int winner, loser;
        cin >> winner >> loser;
        win[winner].push_back(loser);
        lose[loser].push_back(winner);
    }
    for (int i = 1; i <= n; i++) {
        DfsWin(i, i);
        ans[i].second -= cnt;
        Refresh();
        DfsLose(i, i);
        ans[i].first += cnt;
        Refresh();
    }
    if (isCycle) {
        cout << "-1";
    }
    else {
        for (int i = 1; i <= n; i++) {
            cout << ans[i].first << " " << ans[i].second << "\n";
        }
    }
}