結果

問題 No.1640 簡単な色塗り
ユーザー abc3abc3
提出日時 2021-08-07 04:54:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,090 bytes
コンパイル時間 2,158 ms
コンパイル使用メモリ 200,356 KB
最終ジャッジ日時 2025-01-23 16:36:07
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 WA * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

   #include <bits/stdc++.h>
    using namespace std;
    #include <math.h>
    #include <iomanip>
    #include <cstdint>
    #include <string>
    #include <sstream>
    template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
    template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }
    #define rep(i,n) for (int i = 0; i < (n); ++i)
    typedef long long ll;
    typedef unsigned long long ull;
    using P=pair<ll,ll>;
    const ll INF=1e18;
    const int mod=998244353;

    struct Unionfind{
      vector<int>d;
      Unionfind(int n):d(n,-1){}
      int root(int x){
        if(d[x]<0){return x;}
        return d[x]=root(d[x]);
      }
      bool unite(int x,int y){
        x=root(x);y=root(y);
        if(x==y){return false;}
        if(d[x]>d[y]){swap(x,y);}
        d[x]+=d[y];
        d[y]=x;
        return true;
      }
      bool same(int x,int y){return root(x)==root(y);}
      int size(int x){return -d[root(x)];}
    };

    void solve(){
        int n;
        cin>>n;
        Unionfind uf(n);
        vector<int>cnt(n);
        vector<int>a(n),b(n);
        
        vector<int>used(n),ans(n,-1);
        rep(i,n){
            cin>>a[i]>>b[i];
            a[i]--;b[i]--;
            uf.unite(a[i],b[i]);
            if(a[i]==b[i]){ans[i]=a[i]; used[a[i]]=1; }
        }
        rep(i,n){cnt[uf.root(a[i])]++;}
        
        rep(i,n){
            if(uf.root(i)==i){
                if(cnt[uf.root(i)]<uf.size(i)){
                    cout<<"No"<<endl;
                    return;
                }
            }
            if(ans[i]!=-1){continue;}
            if(!used[a[i]]){
                used[a[i]]=1;
                ans[i]=a[i];
            }
            else{
                used[b[i]]=1;
                ans[i]=b[i];
            }
        }
        cout<<"Yes"<<endl;
        rep(i,ans.size()){
            cout<<ans[i]+1<<endl;
        }
    }

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

        return 0;
    }
0