結果

問題 No.74 貯金箱の退屈
ユーザー srup٩(๑`н´๑)۶
提出日時 2016-08-26 20:44:32
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 2,416 bytes
コンパイル時間 852 ms
コンパイル使用メモリ 78,904 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-08 09:28:56
合計ジャッジ時間 1,904 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <set>
#include <cstring>
using namespace std;
typedef long long ll;
#define rep(i,n) for(int i=0;i<(n);i++)

class DisjointSet{
    public:
        vector<int> rank, p;//rank:木の高さ p:親の頂点番号
        DisjointSet(){}
        DisjointSet(int size){//頂点の数
            rank.resize(size, 0);
            p.resize(size, 0);
            rep(i, size) makeSet(i);
        }
        void makeSet(int x){
            p[x] = x;
            rank[x] = 0;
        }
        //同じ木にあるか判定（今回使わない）
        bool same(int x, int y){
            return findSet(x) == findSet(y);
        }
        // 木どうしをくっつける
        void unite(int x, int y){
            link(findSet(x), findSet(y));
        }
        //木の高さを考慮して木どうしをくっつける
        void link(int x, int y){
            if(rank[x] > rank[y]){
                p[y] = x;
            }else{
                p[x] = y;
                if(rank[x] == rank[y]) rank[y]++;
            }
        }
        //親を探す（ルートまで）
        int findSet(int x){
            if(x != p[x]){
                p[x] = findSet(p[x]);
            }
            return p[x];
        }
};

int main(void){
    int n; cin >> n;
    DisjointSet ds = DisjointSet(n);
    vector<int> d(n);
    vector<int> w(n);
    rep(i, n) cin >> d[i];
    rep(i, n) cin >> w[i];
    vector<bool> flag(n, 0);//1つだけで反転できるやつ

    vector<pair<int, int> > memo;
    for (int i = 0; i < n; ++i){
        int l = (i - d[i] + 1000 * n) % n;
        int r = (i + d[i]) % n;
        // printf("%d %d\n", l, r);
        memo.push_back(make_pair(l, r));
        if(l == r) flag[l] = true;
    }

    set<int> s;//木に追加した頂点番号
    for(auto v : memo){
        if(v.first == v.second) continue;
        if(flag[v.first]) continue;
        if(flag[v.second]) continue;
        ds.unite(v.first, v.second);
        s.insert(v.first); s.insert(v.second);
    }

    vector<int> parentura(n, 0); //根がiのもので始めに裏が奇数のものの数
    for (auto v : s){
        int ret = ds.findSet(v);
        if(w[ret] == 0) parentura[ret]++;
    }
    rep(i, n){
        if(parentura[i] % 2 != 0){
            printf("No\n");
            return 0;
        }
    }
    printf("Yes\n");
    return 0;
}
0