結果

問題 No.74 貯金箱の退屈
ユーザー koyopro
提出日時 2015-10-01 13:17:25
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,621 bytes
コンパイル時間 1,348 ms
コンパイル使用メモリ 161,312 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-19 18:52:13
合計ジャッジ時間 1,956 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;

#define REP(i, n) for(int i=0; i<(n); i++)
#define FOR(i, a, b) for(int i=(a);i<(b);i++)

int N;
struct coin { int D, W, U, Single; };
vector<coin> coins(1000);

int top(int i) {
    if (coins[i].U < 0) return coins[i].U;
    if (coins[i].U == i) return i;
    return top(coins[i].U);
}

void setu(int i) {
    int bi = (i + coins[i].D) % N;
    int ai = (N*2000 + i - coins[i].D) % N;
    int bg = top(bi);
    int ag = top(ai);
    if (bg < 0 && ag < 0) {
        // 未所属同士: グループ化
        coins[bi].U = coins[ai].U = max(bi, ai);
    } else if (bg < 0){
        // 未所属の方を連結
        coins[bi].U = ag;
    } else if (ag < 0){
        // 未所属の方を連結
        coins[ai].U = bg;
    } else {
        // グループの先頭を連結
        coins[bg].U = ag;
    }
    if (bi == ai) coins[bi].Single = 1;
}

bool isok(int g) {
    int backcnt = 0;
    int singlecnt = 0;
    REP(j,N) {
        if (top(j) != g) continue;
        coin c = coins[j];
        backcnt += (c.W == 0);
        // 一つを返せるやつかどうか
        singlecnt += c.Single;
    }
    return (singlecnt > 0) || (backcnt % 2 == 0);
}

signed main()
{
    cin >> N;
    REP(i,N) cin >> coins[i].D;
    REP(i,N) cin >> coins[i].W;
    REP(i,N) {
        coins[i].U = -1 * (i+1);
        coins[i].Single = 0;
    }
    REP(i,N) {
        // Union-Find木を構築
        setu(i);
    }
    bool ok = true;
    FOR(i,-1*N,N) {
        // Union-Find木毎に判定
        ok &= isok(i);
    }
    cout << (ok ? "Yes" : "No") << endl;
    return 0;
}
0