結果

問題 No.2301 Namorientation
ユーザー otoshigootoshigo
提出日時 2023-05-12 21:40:19
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 181 ms / 3,000 ms
コード長 1,802 bytes
コンパイル時間 922 ms
コンパイル使用メモリ 90,236 KB
実行使用メモリ 37,616 KB
最終ジャッジ日時 2024-11-28 17:42:10
合計ジャッジ時間 10,642 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<iostream>
#include<vector>
#include<algorithm>
#include<tuple>
using namespace std;
using ll=long long;
#define rep(i,n) for(int i=0;i<n;i++)
#define rrep(i,n) for(int i=(n)-1;i>=0;i--)
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
template<class T> bool chmax(T &a, T b){if (a < b){a = b;return true;} else return false;}
template<class T> bool chmin(T &a, T b){if (a > b){a = b;return true;} else return false;}

struct UnionFind {
    vector<int> par, siz;
    UnionFind(int n) : par(n,-1), siz(n,1) {}
    int find(int x){
        if (par[x] == -1) return x;
        else{
            par[x] = find(par[x]);
            return par[x];
        }
    }
    bool same(int x, int y){
        return find(x) == find(y);
    }
    int size(int x){
        return siz[find(x)];
    }
    void merge(int x, int y){
        x = find(x);
        y = find(y);
        if (x == y) return;
        if (siz[x] < siz[y]) swap(x,y);
        siz[x] += siz[y];
        par[y] = x;
    }
};

void dfs(int x,int p,vector<vector<tuple<int,int,int>>>&g,vector<string>&ans){
    for(auto[i,e,k]:g[x]){
        if(i==p)continue;
        dfs(i,x,g,ans);
        if(k==1)ans[e]="<-";
        else ans[e]="->";
    }
}

int main(){
    ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int n;
    cin>>n;
    vector<vector<tuple<int,int,int>>>g(n);
    UnionFind uf(n);
    int s=-1,t=-1;
    vector<string>ans(n);
    rep(i,n){
        int a,b;
        cin>>a>>b;
        a--;
        b--;
        if(uf.same(a,b)){
            s=a;
            t=b;
            ans[i]="->";
        }else{
            uf.merge(a,b);
            g[a].push_back(make_tuple(b,i,1));
            g[b].push_back(make_tuple(a,i,0));
        }
    }
    dfs(s,-1,g,ans);
    rep(i,n)cout<<ans[i]<<"\n";
}
0