#713782 (C++17) No.1769 Don't Stop the Game

• 問題: No.1769 Don't Stop the Game
• ユーザー: polylogK
• 提出日時: 2021-11-06 13:45:53
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 282 ms / 3,000 ms
• コード長: 1,235 bytes
• コンパイル時間: 595 ms
• コンパイル使用メモリ: 78,176 KB
• 実行使用メモリ: 62,348 KB
• 最終ジャッジ日時: 2024-06-29 18:03:38
• 合計ジャッジ時間: 5,891 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 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	105 ms; 16,256 KB
testcase_09	AC	78 ms; 13,696 KB
testcase_10	AC	157 ms; 24,156 KB
testcase_11	AC	67 ms; 12,800 KB
testcase_12	AC	120 ms; 19,288 KB
testcase_13	AC	125 ms; 19,456 KB
testcase_14	AC	129 ms; 19,456 KB
testcase_15	AC	143 ms; 19,456 KB
testcase_16	AC	146 ms; 19,688 KB
testcase_17	AC	182 ms; 22,140 KB
testcase_18	AC	226 ms; 30,896 KB
testcase_19	AC	272 ms; 37,128 KB
testcase_20	AC	282 ms; 37,348 KB
testcase_21	AC	272 ms; 37,500 KB
testcase_22	AC	257 ms; 37,364 KB
testcase_23	AC	125 ms; 19,456 KB
testcase_24	AC	124 ms; 19,568 KB
testcase_25	AC	69 ms; 20,044 KB
testcase_26	AC	178 ms; 38,064 KB
testcase_27	AC	95 ms; 44,416 KB
testcase_28	AC	234 ms; 62,348 KB
testcase_29	AC	218 ms; 49,616 KB
testcase_30	AC	221 ms; 49,772 KB

ソースコード

#include <stdio.h>
#include <algorithm>
#include <vector>
#include <unordered_map>

int main() {
	int n; scanf("%d", &n);
	std::vector<std::vector<int>> g(n);
	std::vector<int> a(n - 1), b(n - 1), c(n - 1);
	for(int i = 0; i < n - 1; i++) {
		scanf("%d%d%d", &a[i], &b[i], &c[i]); a[i]--; b[i]--;

		g[a[i]].push_back(i);
		g[b[i]].push_back(i);
	}

	using i64 = long long;

	i64 ans = (i64)n * (n - 1);
	std::vector<int> x(n), root(n), size(n), count(n);
	std::unordered_map<int, int> count_root; {
		std::unordered_map<int, int> map;
		auto dfs = [&](auto&& dfs, int v, int par, int xor_val) -> void {
			auto [it, _] = map.try_emplace(xor_val, -1);
			int tmp = it->second;

			if(tmp == -1) count_root[xor_val]++;
			else count[tmp]++;

			x[v] = xor_val;
			root[v] = tmp;
			size[v] = 1;
			for(int id: g[v]) {
				int to = a[id] ^ b[id] ^ v;
				if(to == par) continue;

				map[xor_val] = to;
				dfs(dfs, to, v, xor_val ^ c[id]);

				size[v] += size[to];
				ans -= (i64)count[to] * (n - size[to]);
			}
			map[xor_val] = tmp;
		}; dfs(dfs, 0, -1, 0);
	}
	for(int v = 0; v < n; v++) {
		if(root[v] == -1) ans -= (i64)(count_root[x[v]] - 1) * size[v];
		else ans -= (i64)count[root[v]] * size[v];
	}
	printf("%lld\n", ans);
}