結果
問題 | No.74 貯金箱の退屈 |
ユーザー |
![]() |
提出日時 | 2020-05-11 16:32:57 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 5,000 ms |
コード長 | 1,626 bytes |
コンパイル時間 | 1,475 ms |
コンパイル使用メモリ | 178,768 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-07-18 13:12:16 |
合計ジャッジ時間 | 2,474 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#include <bits/stdc++.h> using namespace std; struct UnionFind { vector<int> data; UnionFind(int n) { data.assign(n, -1); } bool unite(int x, int y) { x = root(x), y = root(y); if (x == y) return (false); if (data[x] > data[y]) swap(x, y); data[x] += data[y]; data[y] = x; return (true); } int root(int k) { if (data[k] < 0) return (k); return (data[k] = root(data[k])); } bool same(int x, int y) { return root(x) == root(y); } int size(int k) { return (-data[root(k)]); } }; int rem(int a, int b) { return ((a %= b) += b) %= b; } int main() { int N, D[110], W[110]; cin >> N; for (int i = 0; i < N; i++) { cin >> D[i + 1]; } for (int i = 0; i < N; i++) { cin >> W[i + 1]; } int ok[110] = {}; UnionFind uf(N + 1); for (int i = 1; i <= N; i++) { uf.unite(rem(i + D[i] - 1, N) + 1, rem(i - D[i] - 1, N) + 1); if (rem(i + D[i] - 1, N) + 1 == rem(i - D[i] - 1, N) + 1) { ok[rem(i + D[i] - 1, N) + 1] = 1; } } map<int, vector<int>> mp; for (int i = 1; i <= N; i++) { mp[uf.root(i)].push_back(i); } bool ans = true; for (auto& p : mp) { auto& V = p.second; int cnt = 0; for (auto& x : V) { cnt += !W[x]; if (ok[x]) { cnt = 0; break; } } if (cnt & 1) ans = false; } cout << (ans ? "Yes" : "No") << endl; }