結果

問題 No.74 貯金箱の退屈
ユーザー 🍮かんプリン🍮かんプリン
提出日時 2019-09-11 22:15:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,873 bytes
コンパイル時間 1,674 ms
コンパイル使用メモリ 181,168 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-02 16:53:56
合計ジャッジ時間 2,635 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

#include "bits/stdc++.h"
#define ALL(obj) (obj).begin(),(obj).end()
#define RALL(obj) (obj).rbegin(),(obj).rend()
#define REP(i, n) for(int i = 0; i < (int)(n); i++)
#define REPR(i, n) for(int i = (int)(n); i >= 0; i--)
#define FOR(i,n,m) for(int i = (int)(n); i < int(m); i++)
using namespace std;
typedef long long ll;
const int MOD = 1e9 + 7;
const int INF = MOD - 1;
const ll LLINF = 4e18;

//UnionFind
class UnionFind {
private:
    vector<int> par;
public:
    UnionFind(int n) {
        par.resize(n, -1);
    }

    int root(int x) {
        if (par[x] < 0) return x;
        return par[x] = root(par[x]);
    }

    bool unite(int x, int y) {
        int rx = root(x);
        int ry = root(y);
        if (rx == ry) return false;
        if (size(rx) < size(ry)) swap(rx, ry);
        par[rx] += par[ry];
        par[ry] = rx;
        return true;
    }

    bool same(int x, int y) {
        int rx = root(x);
        int ry = root(y);
        return rx == ry;
    }

    int size(int x) {
        return -par[root(x)];
    }
};

int main() {
    int n; cin >> n;
    UnionFind uni(n);
    vector<int> d(n),w(n);
    vector<int> b;
    REP(i, n) {
        cin >> d[i];
        d[i] -= d[i] / n * n;
    }
    REP(i, n) cin >> w[i];
    REP(i, n) {
        int a = (i + d[i]) % n;
        int c = (n + i - d[i]) % n;
        if (a == c) {
            b.push_back(a);
        }
        else {
            uni.unite(a, c);
        }
    }
    map<int, int> mp;
    REP(i, n) {
        mp[uni.root(i)] = 0;
    }
    for (auto e : b) {
        mp[uni.root(e)] = -1;
    }
    REP(i, n) {
        if (w[i] == 0 && mp[uni.root(i)] != -1) {
            mp[uni.root(i)]++;
        }
    }
    for (auto e : mp) {
        if (e.second != -1 && e.second % 2 == 1) {
            puts("No");
            return 0;
        }
    }
    puts("Yes");
    getchar(); getchar();
}
0