#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}; } 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, int> color; range(i, 0, n-1){ int a, b, c;std::cin >> a >> b >> c; a--, b--, c--; G[a].push_back(b); G[b].push_back(a); if(a > b) swap(a, b); color[pair{a, b}] = c; } auto [g, root, parent, depth] = centroid_decomposition(G); //cから初めて色が2色以下の間探す ll ans = 0; for(int i=0;i, int> ALL; ll O = 0; vector, int>> subtree; vector Ocount; for(int ch:G[i]){ subtree.push_back(map, int>()); Ocount.push_back(0); int idx = subtree.size() - 1; if(depth[ch]> q; q.push({ch, i, color[pair{min(i, ch), max(i, ch)}], -1}); while(!q.empty()){ auto [now, from, c1, c2] = q.front(); q.pop(); if(c2==-1){ O++, Ocount[idx]++; ALL[pair{c1, c2}]++; subtree[idx][pair{c1, c2}]++; }else{ assert(c1!=c2); ALL[pair{min(c1, c2), max(c1, c2)}]++; subtree[idx][{min(c1, c2), max(c1, c2)}]++; } for(auto nxt:G[now]){ if(nxt==from||depth[nxt]<=depth[i]) continue; int ecolor = color[pair{min(now, nxt), max(now, nxt)}]; if(c1==ecolor) q.push({nxt, now, c1, c2}); else if(c2==-1||c2==ecolor) q.push({nxt, now, c1, ecolor}); } } } for(int j=0;j{c1, -1}]; int except_c1_i = Ocount[j] - subtree[j][pair{c1, -1}]; ans += ll(except_c1_all - except_c1_i) * count; //std::cout << "add: " << c1 << " " << -1 << " " << ll(except_c1_all - except_c1_i) * count << '\n'; }else{ ans += count * 2; //std::cout << "add: " << c1 << " " << c2 << " " << count * 2 << '\n'; ans += ll(ALL[pair{c1, c2}] - count) * count; } } } } assert(ans%2==0); std::cout << ans / 2 << '\n'; }