結果

問題 No.2301 Namorientation
ユーザー k1suxuk1suxu
提出日時 2023-05-12 23:45:32
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,875 bytes
コンパイル時間 3,687 ms
コンパイル使用メモリ 268,880 KB
実行使用メモリ 51,856 KB
最終ジャッジ日時 2024-05-06 13:41:39
合計ジャッジ時間 19,220 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 WA -
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 AC 311 ms
51,856 KB
testcase_23 AC 317 ms
51,084 KB
testcase_24 AC 261 ms
43,208 KB
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

// #pragma GCC target("avx")
// #pragma GCC optimize("O3")
// #pragma GCC optimize("unroll-loops")

#include <bits/stdc++.h>
using namespace std;

#define rep(i,n) for(int i = 0; i < (int)n; i++)
#define FOR(n) for(int i = 0; i < (int)n; i++)
#define repi(i,a,b) for(int i = (int)a; i < (int)b; i++)
#define all(x) x.begin(),x.end()
//#define mp make_pair
#define vi vector<int>
#define vvi vector<vi>
#define vvvi vector<vvi>
#define vvvvi vector<vvvi>
#define pii pair<int,int>
#define vpii vector<pair<int,int>>

template<typename T>
void chmax(T &a, const T &b) {a = (a > b? a : b);}
template<typename T>
void chmin(T &a, const T &b) {a = (a < b? a : b);}

using ll = long long;
using ld = long double;
using ull = unsigned long long;

const ll INF = numeric_limits<long long>::max() / 2;
const ld pi = 3.1415926535897932384626433832795028;
const ll mod = 998244353;
int dx[] = {1, 0, -1, 0, -1, -1, 1, 1};
int dy[] = {0, 1, 0, -1, -1, 1, -1, 1};

#define int long long

template< typename G >
struct Namori
{
    const G &g;
    vector< int > in;

    Namori(const G &g) : g(g), in(g.size(), 0) {}

    void decomposition(vector< int > &loop, vector< vector< int > > &forest)
    {
        int N = (int) in.size();
        for(int i = 0; i < N; i++) {
            in[i] = g[i].size();
        }
        forest.resize(N);
        queue< int > que;
        vector< bool > v(N, 0);
        for(int i = 0; i < N; i++) {
            if(in[i] == 1) {
                que.emplace(i);
                v[i] = true;
            }
        }
        while(!que.empty()) {
            int idx = que.front();
            que.pop();
            for(auto &to : g[idx]) {
                if(v[to.first]) continue;
                --in[to.first];
                forest[to.first].push_back(idx);
                forest[idx].push_back(to.first);
                if(in[to.first] > 1) continue;
                que.emplace(to.first);
                v[to.first] = true;
            }
        }

        function< void(int) > dfs = [&](int idx) {
            loop.push_back(idx);
            for(auto &to : g[idx]) {
                if(v[to.first]) continue;
                v[to.first] = true;
                dfs(to.first);
            }
        };
        for(int i = 0; i < N; i++) {
            if(v[i]) continue;
            v[i] = true;
            dfs(i);
            break;
        }
    }
};

void solve() {
    int n;
    cin >> n;
    vi a(n), b(n);
    vector<vpii> g(n);
    FOR(n) {
        cin >> a[i] >> b[i];
        --a[i];
        --b[i];
        g[a[i]].emplace_back(b[i], i);
        g[b[i]].emplace_back(a[i], i);
    }
    Namori<vector<vpii>> namori(g);
    vi loop;
    vvi forest;
    namori.decomposition(loop, forest);

    vector<bool> in_loop(n, false);
    for(auto e : loop) in_loop[e] = true;
    vi ans(n);
    vector<bool> done(n, false);

    auto dfs = [&](int v, int p, auto self) -> void {
        done[v] = true;
        bool flag = false;
        for(auto e : g[v]) if(!done[e.first] && e.first != p && in_loop[e.first]) {
            ans[e.second] = v;
            flag = true;
            self(e.first, v, self);
        }
        if(!flag) {
            for(auto e : g[v]) if(e.first != p) {
                ans[e.second] = v;
                return;
            }
        }
    };
    dfs(loop[0], -1, dfs);

    vi end;
    FOR(n) if(g[i].size() == 1) end.push_back(i);

    for(int now : end) {
        while(!in_loop[now]) {
            in_loop[now] = true;
            for(auto e : g[now]) if(!in_loop[e.first]) {
                ans[e.second] = now;
                now = e.first;
                break;
            }
        }
    }

    FOR(n) {
        if(ans[i] == a[i]) cout << "->" << endl;
        else cout << "<-" << endl;
    }
}

signed main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    solve();
    return 0;
}
0