結果

問題 No.1640 簡単な色塗り
ユーザー riano
提出日時 2021-08-06 22:51:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 3,435 bytes
コンパイル時間 2,134 ms
コンパイル使用メモリ 188,824 KB
実行使用メモリ 17,280 KB
最終ジャッジ日時 2024-06-29 15:56:50
合計ジャッジ時間 10,882 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31 WA * 17 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define rep(i,n) for(int (i)=0;(i)<(n);(i)++)
#define Pr pair<ll,ll>
#define Tp tuple<ll,ll,ll>
using Graph = vector<vector<int>>;
#define double long double
const ll mod = 1000000007;
//Unionfind
struct unionfind {
    //自身が親であれば、その集合に属する頂点数に-1を掛けたもの
    //そうでなければ親のid
      vector<long long> r;
      vector<bool> act;
      unionfind(long long N) {
            r = vector<long long>(N, -1);
            act = vector<bool>(N,false);
      }
      long long root(long long x) {
            if (r[x] < 0) return x;
            return r[x] = root(r[x]);
      }
      bool unite(long long x, long long y) {
            x = root(x);
            y = root(y);
            if (x == y) return false;
            if (r[x] > r[y]) swap(x, y);
            r[x] += r[y];
            r[y] = x;
            return true;
      }
      long long size(long long x) {
            return max(-r[root(x)],0LL);
      }
  
  // 2つのデータx, yが属する木が同じならtrueを返す
    bool same(long long x, long long y) { 
        long long rx = root(x);
        long long ry = root(y);
        if(rx==ry){
            act[rx] = true;
        }
        return rx == ry;
    }
    bool activate(ll x){
        long long rx = root(x);
        return act[rx];
    }
};
  //main関数内で unionfind friends(N+1);
  //などと宣言し  friends.unite(a,b);
  //のように使う
int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    ll N; cin >> N;
    ll a[N],b[N];
    ll cnt[N+1];
    rep(i,N) cnt[i+1] = 0;
    Graph G(N+1);
    rep(i,N){
        cin >> a[i] >> b[i];
        G[a[i]].push_back(i);
        G[b[i]].push_back(i);
        cnt[a[i]]++; cnt[b[i]]++;
    }
    unionfind balls(N+1);
    rep(i,N){
        if(!balls.same(a[i],b[i])){
            balls.unite(a[i],b[i]);
        }
    }
    rep(i,N){
        if(!balls.activate(i+1)){
            cout << "No" << endl; return 0;
        }
    }
    cout << "Yes" << endl;
    set<ll> used,app;
    ll ans[N];
    rep(i,N) ans[i] = -1;
    queue<ll> go;
    rep(i,N){
        if(cnt[i+1]==1){
            go.push(i+1);
        }
    }
    rep(i,N){
        app.insert(i+1);
    }
    ll cn = 0;
    while(1){
        if(cn!=N&&go.empty()){
            auto itr = app.begin();
            ll t = *itr;
            app.erase(itr);
            ll y = -1;
            
            for(ll x:G[t]){
                if(!used.count(x)){
                    y = x; break;
                }
            }
            while(y==-1){
                ll ss = 0;
            }
            //cout << t << " " << y << endl;
            ans[y] = t;
            cnt[a[y]]--;
            cnt[b[y]]--;
            cnt[t] = -1;
            used.insert(y);
            if(cnt[a[y]]==1&&a[y]!=t){
                go.push(a[y]);
            }
            if(cnt[b[y]]==1&&b[y]!=t){
                go.push(b[y]);
            }
            cn++;
        }
        ll s = go.front(); go.pop();
        for(ll x:G[s]){
            if(used.count(x)) continue;
            ans[x] = s; app.erase(s);
            cnt[a[x]]--; cnt[b[x]]--;
            cnt[s] = -1;
            if(cnt[a[x]]==1){
                go.push(a[x]);
            }
            if(cnt[b[x]]==1){
                go.push(b[x]);
            }
            used.insert(x); cn++;
        }
        if(cn==N) break;
    }
    rep(i,N){
        cout << ans[i] << endl;
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0