/* # Algorithm 各頂点 v と、その親 p について次の値を求める。 c[v] := v の部分木にあってかつ xor[p; u] がはじめて 0 になるような u の個数 ## Time Comlexity O(N) */ #include #include #include #include int main() { int n; scanf("%d", &n); std::vector> g(n); std::vector 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 = 0; std::vector x(n), root(n), size(n); std::vector count(n); std::unordered_map count_root; { std::unordered_map 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 += 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 += (count_root[x[v]] - 1) * size[v]; else ans += count[root[v]] * size[v]; } ans = (i64)n * (n - 1) - ans; printf("%lld\n", ans); }