#include using namespace std; vector> edges[200000]; bitset<200000> dead; struct Centroid{ vector centroid; int sz[200000]; int all_sz; int get_size(int i, int prev){ int sz = 1; for(auto& e : edges[i]){ int j = e.first; if(j == prev || dead[j]) continue; sz += get_size(j, i); } return sz; } void dfs(int i, int prev){ sz[i] = 1; bool is_centroid = true; for(auto& e : edges[i]){ int j = e.first; if(j == prev || dead[j]) continue; dfs(j, i); sz[i] += sz[j]; if(sz[j] > all_sz/2) is_centroid = false; } if(all_sz - sz[i] > all_sz/2) is_centroid = false; if(is_centroid) centroid.push_back(i); } int calculate(int s){ centroid.clear(); all_sz = get_size(s, -1); dfs(s, -1); assert(centroid.size() > 0); return centroid[0]; } }; int64_t ans = 0; Centroid cent; int dfs1(int i, int p, int c){ int res = 1; for(auto& e : edges[i]){ int j = e.first, col = e.second; if(j == p || dead[j]) continue; if(c == col) res += dfs1(j, i, c); } return res; } void dfs2(int i, int p, int c1, int c2, map, int>& mp){ if(c2 != -1) mp[minmax(c1, c2)]++; for(auto& e : edges[i]){ int j = e.first, col = e.second; if(j == p || dead[j]) continue; if(c2 != -1 && col != c1 && col != c2) continue; if(c1 == col){ dfs2(j, i, c1, c2, mp); }else{ dfs2(j, i, c1, col, mp); } } } int single_V[200000], single_C[200000]; void solve(int s){ int c = cent.calculate(s); int64_t single_sum = 0; for(auto& e : edges[c]){ int j = e.first, col = e.second; if(dead[j]) continue; int res = dfs1(j, c, col); single_V[j] = res; single_C[col] += res; single_sum += res; ans += res * (single_sum - single_C[col]); } map, int> mp; for(auto& e : edges[c]){ int j = e.first, col = e.second; if(dead[j]) continue; single_C[col] -= single_V[j]; map, int> mp2; dfs2(j, c, col, -1, mp2); for(auto& pp : mp2){ auto& p = pp.first; int64_t num = 1 + single_C[p.first] + single_C[p.second]; if(mp.count(p)) num += mp[p]; ans += pp.second * num; } for(auto& pp : mp2) mp[pp.first] += pp.second; single_C[col] += single_V[j]; } for(auto& e : edges[c]){ int j = e.first, col = e.second; single_V[j] = single_C[col] = 0; } dead[c] = true; for(auto& e : edges[c]) if(!dead[e.first]) solve(e.first); } int main(){ int N, K; cin >> N >> K; for(int i=0; i> a >> b >> c; a--; b--; c--; edges[a].emplace_back(b, c); edges[b].emplace_back(a, c); } solve(0); cout << ans << endl; return 0; }