#72943 (C++11) No.74 貯金箱の退屈

• 問題: No.74 貯金箱の退屈
• ユーザー: hotpepsi
• 提出日時: 2016-01-27 00:19:11
• 言語: C++11 (gcc 11.4.0)
• 結果: WA
• 実行時間: -
• コード長: 1,427 bytes
• コンパイル時間: 486 ms
• コンパイル使用メモリ: 60,380 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-21 17:38:30
• 合計ジャッジ時間: 1,451 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	1 ms; 5,376 KB
testcase_14	AC	1 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	1 ms; 5,376 KB
testcase_18	WA	-
testcase_19	AC	2 ms; 5,376 KB
testcase_20	AC	1 ms; 5,376 KB
testcase_21	AC	1 ms; 5,376 KB
testcase_22	WA	-
testcase_23	AC	1 ms; 5,376 KB
testcase_24	WA	-
testcase_25	AC	1 ms; 5,376 KB
testcase_26	AC	2 ms; 5,376 KB
testcase_27	AC	2 ms; 5,376 KB
testcase_28	WA	-
testcase_29	WA	-
testcase_30	AC	1 ms; 5,376 KB
testcase_31	AC	2 ms; 5,376 KB
testcase_32	WA	-

ソースコード

#include <iostream>
#include <vector>

using namespace std;

struct UnionFind {
	std::vector<int> _parent;
	std::vector<int> _size;
	UnionFind(int size) : _parent(size, -1), _size(size, 1) { }
	void unite(int a, int b) {
		int ra = root(a), rb = root(b); if (ra == rb) { return; }
		if (_size[ra] >= _size[rb]) { _parent[rb] = ra, _size[ra] += _size[rb]; } else { _parent[ra] = rb, _size[rb] += _size[ra]; }
	}
	int root(int a) {
		int p = _parent[a];
		if (p < 0) { return a; }
		while (_parent[p] >= 0) { p = _parent[p]; }
		return _parent[a] = p;
	}
	int size(int a) { return _size[root(a)]; }
};

int main(int argc, char *argv[])
{
	int N;
	cin >> N;
	int tot = 0;
	UnionFind uf(N);
	int D[256], W[256], OK[256] = {}, CNT[256] = {};
	for (int i = 0; i < N; ++i) {
		cin >> D[i];
	}
	for (int i = 0; i < N; ++i) {
		cin >> W[i];
		tot += W[i];
	}
	for (int i = 0; i < N; ++i) {
		int a = (i - (D[i] % N) + N) % N, b = (i + D[i]) % N;
		uf.unite(a, b);
	}
	for (int i = 0; i < N; ++i) {
		int a = (i - (D[i] % N) + N) % N, b = (i + D[i]) % N;
		if (a == b) {
			OK[uf.root(i)] = 1;
		}
		CNT[uf.root(i)] += W[i];
	}
	bool ans = true;
	for (int i = 0; i < N; ++i) {
		if (uf.root(i) == i && uf.size(i) > 1) {
			int sz = uf.size(i);
			int cnt = CNT[i];
			if (!OK[i] && (uf.size(i) - CNT[i]) % 2) {
				ans = false;
				break;
			}
		}
	}
	if (tot >= N) {
		ans = true;
	}
	cout << (ans ? "Yes" : "No") << endl;
	return 0;
}