#ifndef LOCAL #pragma GCC optimize("O3,no-stack-protector") #pragma GCC target("arch=x86-64-v3") #endif #include using namespace std; using ll = long long; namespace std { template struct hash> { size_t operator()(pair const &val) const { return hash{}(val.first)*1337^hash{}(val.second); }};} struct CentroidDecomposition { int n, r; vector> g, cd; vector used, ord, szs; CentroidDecomposition(vector> &gg): g(gg), n(gg.size()), used(n), cd(n), szs(n) { dfs1(0,-1); dfs(0,-1,n,-1); } int dfs1(int v, int p) { if(szs[v]==-1) return 0; szs[v]=1; for(auto e:g[v])if(e!=p)szs[v]+=dfs1(e,v); return szs[v]; } int dfs(int node, int p, int sz, int root){ if(szs[node]==-1) return false; bool b=false; int res=1; for(auto e:g[node]){ if (p==e || szs[e]<=sz/2) continue; if (dfs(e,node,sz,root)) return true; b=true; } if(b) return false; if(root!=-1)cd[root].emplace_back(node); else r=node; ord.push_back(node); dfs1(node,-1); szs[node]=-1; for(auto e:g[node])dfs(e,node,szs[e],node); return true; } }; bool DEBUG = false; using pii = pair; int cnt0 = 0; // *, - int cnt1 = 0; // *, * map cnt2; // x, y map cnt3; // x, * map cnt4; // x, - vector> g; vector> edges; vector visited; void f1(int node, int parent, int c1, int c2) { if (c2 == -1) { cnt0++; cnt4[c1]++; } else { cnt1++; cnt2[make_pair(min(c1, c2), max(c1, c2))]++; cnt3[c1]++; cnt3[c2]++; } for (auto [child, c] : edges[node]) { if (child == parent || visited[child]) continue; if (c1 == c) { f1(child, node, c1, c2); } else if (c2 == c || c2 == -1) { f1(child, node, c1, c); } } }; ll ans; void f2(int node, int parent, int c1, int c2) { if (c2 == -1) { ans += cnt0 - cnt4[c1] + cnt3[c1]; } else { ans += cnt2[make_pair(min(c1, c2), max(c1, c2))] + cnt4[c1] + cnt4[c2]; } for (auto [child, c] : edges[node]) { if (child == parent || visited[child]) continue; if (c1 == c) { f2(child, node, c1, c2); } else if (c2 == c || c2 == -1) { f2(child, node, c1, c); } } }; int main() { auto now = [](){ return chrono::system_clock::now(); }; auto start = now(); auto cur = [&](){ return (now()-start).count() / 1000'000.; }; cin.tie(0)->sync_with_stdio(false); int n, k; cin >> n >> k; g = vector>(n); edges = vector>(n); for (int i = 0; i < n - 1; i++) { int u, v, c; cin >> u >> v >> c; u--, v--, c--; g[u].push_back(v); g[v].push_back(u); edges[u].emplace_back(v, c); edges[v].emplace_back(u, c); } if (DEBUG) cout << cur() << endl; CentroidDecomposition cd(g); for (int i = 1; i < 0; i++) { cd = CentroidDecomposition(g); } if (DEBUG) cout << cur() << endl; visited = vector(n); ans = 0; for (int ce : cd.ord) { cnt0=cnt1=0; cnt2.clear(); cnt3.clear(); cnt4.clear(); for (auto [child, c] : edges[ce]) { if (visited[child]) continue; f2(child, ce, c, -1); f1(child, ce, c, -1); } ans += cnt1; visited[ce] = true; } if (DEBUG) cout << cur() << endl; cout << ans << endl; }