#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #define all(obj) (obj).begin(), (obj).end() #define range(i, l, r) for(int i=l;i>1)|y)) #define bit_kth(i, k) ((i >> k)&1) #define bit_highest(i) (i?63-__builtin_clzll(i):-1) #define bit_lowest(i) (i?__builtin_ctzll(i):-1) using ll = long long; using ld = long double; using ul = uint64_t; using namespace std; template std::ostream &operator<<(std::ostream &dest, const std::pair &p){ dest << p.first << ' ' << p.second; return dest; } template std::ostream &operator<<(std::ostream &dest, const std::vector> &v){ int sz = v.size(); if(sz==0) return dest; for(int i=0;i std::ostream &operator<<(std::ostream &dest, const std::vector &v){ int sz = v.size(); if(sz==0) return dest; for(int i=0;i std::ostream &operator<<(std::ostream &dest, const std::array &v){ if(sz==0) return dest; for(int i=0;i std::ostream &operator<<(std::ostream &dest, const std::set &v){ for(auto itr=v.begin();itr!=v.end();){ dest << *itr; itr++; if(itr!=v.end()) dest << ' '; } return dest; } template std::ostream &operator<<(std::ostream &dest, const std::map &v){ for(auto itr=v.begin();itr!=v.end();){ dest << '(' << itr->first << ", " << itr->second << ')'; itr++; if(itr!=v.end()) dest << '\n'; } return dest; } template vector make_vec(size_t sz, T val){return std::vector(sz, val);} template auto make_vec(size_t sz, Tail ...tail){ return std::vector(tail...))>(sz, make_vec(tail...)); } template vector read_vec(size_t sz){ std::vector v(sz); for(int i=0;i> v[i]; return v; } template auto read_vec(size_t sz, Tail ...tail){ auto v = std::vector(tail...))>(sz); for(int i=0;i(tail...); return v; } template bool mul_overflow(T a, T b){ static T INF = numeric_limits::max(); return (INF / a) < b; } template bool add_overflow(T a, T b){ static T INF = numeric_limits::max(); return (INF - a) < b; } template bool chmax(T &a, const T b){ if(a >= b) return false; a = b; return true; } template bool chmin(T &a, const T b){ if(a <= b) return false; a = b; return true; } template T max(vector &v, int l=0, int r = -1){ return *std::max_element(v.begin()+l, v.begin()+(r==-1?v.size():r)); } template T min(vector &v, int l=0, int r = -1){ return *std::min_element(v.begin()+l, v.begin()+(r==-1?v.size():r)); } template int argmax(vector &v, int l=0, int r=-1){ T res = max(v, l, r); if(r==-1) r = v.size(); for(int i=l;i int argmin(vector &v, int l=0, int r=-1){ T res = min(v, l, r); if(r==-1) r = v.size(); for(int i=l;i>= __builtin_ctzll(a); do{ b >>= __builtin_ctzll(b); if(a > b) std::swap(a, b); b -= a; }while(b); return (a << shift); } ll lcm(ll a, ll b){ return (a / gcd(a, b)) * b; } ll llpow(ll a, ll b){ ll ret = 1, mul = a; while(b){ if(b&1) ret *= mul; mul *= mul; b >>= 1; } return ret; } template ll mpow(ll a, ll b){ int ret = (MOD==1?0:1), mul = a % MOD; while(b){ if(b&1) ret = (ll(ret) * mul) % MOD; mul = (ll(mul) * mul) % MOD; b >>= 1; } return ret; } ll mpow(ll a, ll b, ll MOD){ int ret = (MOD==1?0:1), mul = a % MOD; while(b){ if(b&1) ret = (ll(ret) * mul) % MOD; mul = (ll(mul) * mul) % MOD; b >>= 1; } return ret; } int centroid_search(int now, int from, const vector> &G, const vector &is_centroid, vector &size, int th, vector &parent){ int c = -1; parent[now] = from; size[now] = 1; for(int to:G[now]){ if(to==from||is_centroid[to]) continue; int tmp_c = centroid_search(to, now, G, is_centroid, size, th, parent); size[now] += size[to]; if(tmp_c!=-1) c = tmp_c; } if(c==-1&&size[now]>th) return now; return c; } tuple>, int, vector, vector> centroid_decomposition(const vector> &G){ int n = G.size(), root; vector is_centroid(n, false); vector size(n), parent(n), depth(n), inner_parent(n); vector> g(n); queue> q; q.push({0, -1, n}); while(!q.empty()){ auto [now, par, subtree_size] = q.front(); q.pop(); if(is_centroid[now]) continue; int c = centroid_search(now, -1, G, is_centroid, size, subtree_size/2, inner_parent); parent[c] = par; depth[c] = (par==-1?0:depth[par] + 1); if(par==-1) root = c; else g[par].push_back(c); is_centroid[c] = true; for(int to:G[c]){ if(is_centroid[to]) continue; while(inner_parent[to]!=-1&&inner_parent[to]!=c) to = inner_parent[to]; q.push({to, c, size[to]>size[c]?size[to]-size[c]:size[to]}); } } return {g, root, parent, depth}; } void map_set(unordered_map &mp, int a, int b, int c){ if(b!=0&&a > b) swap(a, b); auto itr = mp.find(ll(a)*1000000+b); if(itr==mp.end()) mp.emplace(ll(a)*1000000+b, c); else itr->second = c; } void map_add(unordered_map &mp, int a, int b){ if(b!=0&&a>b) swap(a, b); auto itr = mp.find(ll(a)*1000000+b); if(itr==mp.end()) mp.emplace(ll(a)*1000000+b, 1); else itr->second++; } int map_get(unordered_map &mp, int a, int b){ if(b!=0&&a>b) swap(a, b); auto itr = mp.find(ll(a)*1000000+b); if(itr==mp.end()) return 0; return itr->second; } using pi = pair; int main(){ cin.tie(nullptr); ios::sync_with_stdio(false); //各重心ノードcでv-cパスに含まれる色を持っておく //cでマージする時、同じ2色, 1色とその色を含む2色, 同じ1色をマージ int n, m;std::cin >> n >> m; vector> G(n); map color; range(i, 0, n-1){ int a, b, c;std::cin >> a >> b >> c; a--, b--; G[a].push_back(b); G[b].push_back(a); color.emplace(pi{min(a, b), max(a, b)}, c); } auto [g, root, parent, depth] = centroid_decomposition(G); //cから初めて色が2色以下の間探す ll ans = 0; ll add = 0; for(int i=0;i ALL; ll O = 0; vector> subtree; vector Ocount; for(int ch:G[i]){ if(depth[ch]<=depth[i]) continue; subtree.push_back(map()); Ocount.push_back(0); int idx = subtree.size() - 1; queue> q; auto itr = color.find(pi{min(i, ch), max(i, ch)}); q.push(tuple{ch, i, itr->second, 0}); while(!q.empty()){ auto [now, from, c1, c2] = q.front(); q.pop(); if(c2==0){ O++, Ocount[idx]++; itr = ALL.find(pi{c1, c2}); if(itr==ALL.end()) ALL.emplace(pi{c1, c2}, 1); else itr->second++; itr = subtree[idx].find(pi{c1, c2}); if(itr==subtree[idx].end()) subtree[idx].emplace(pi{c1, c2}, 1); else itr->second++; }else{ itr = ALL.find(pi{min(c1, c2), max(c1, c2)}); if(itr==ALL.end()) ALL.emplace(pi{min(c1, c2), max(c1, c2)}, 1); else itr->second++; itr = subtree[idx].find(pi{min(c1, c2), max(c1, c2)}); if(itr==subtree[idx].end()) subtree[idx].emplace(pi{min(c1, c2), max(c1, c2)}, 1); else itr->second++; } for(auto nxt:G[now]){ if(nxt==from||depth[nxt]<=depth[i]) continue; int ecolor = color.find(pi{min(now, nxt), max(now, nxt)})->second; if(c1==ecolor) q.push({nxt, now, c1, c2}); else if(c2==0||c2==ecolor) q.push({nxt, now, c1, ecolor}); } } } for(int j=0;jsecond); itr = subtree[j].find(pi{c1, 0}); int except_c1_j = Ocount[j] - (itr==subtree[j].end()?0:itr->second); ans += ll(except_c1_all - except_c1_j) * count; }else{ auto itr = ALL.find(pi{min(c1, c2), max(c1, c2)}); ans += count * 2;//2-0 ans += ll((itr==ALL.end()?0:itr->second) - count) * count; itr = ALL.find(pi{c1, 0}); int c1_all = (itr==ALL.end()?0:itr->second); itr = subtree[j].find(pi{c1, 0}); int c1_j = (itr==subtree[j].end()?0:itr->second); itr = ALL.find(pi{c2, 0}); int c2_all = (itr==ALL.end()?0:itr->second); itr = subtree[j].find(pi{c2, 0}); int c2_j = (itr==subtree[j].end()?0:itr->second); add += ll(c1_all - c1_j) * count; add += ll(c2_all - c2_j) * count; } } } } //std::cout << add << '\n'; std::cout << (ans / 2) + add << '\n'; }