結果

問題 No.1640 簡単な色塗り
コンテスト
ユーザー riano
提出日時 2021-08-06 22:42:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 3,313 bytes
コンパイル時間 1,936 ms
コンパイル使用メモリ 189,116 KB
実行使用メモリ 17,280 KB
最終ジャッジ日時 2024-06-29 15:46:45
合計ジャッジ時間 10,643 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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;
            
            for(ll x:G[t]){
                if(!used.count(x)){
                    y = x; break;
                }
            }
            //cout << t << " " << y << endl;
            ans[y] = t;
            cnt[a[y]]--;
            cnt[b[y]]--;
            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]]--;
            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;
    }
}
0