#include using namespace std; using lint = long long int; using pint = pair; using plint = pair; struct fast_ios { fast_ios(){ cin.tie(0); ios::sync_with_stdio(false); cout << fixed << setprecision(20); }; } fast_ios_; #define ALL(x) (x).begin(), (x).end() #define SZ(x) ((lint)(x).size()) #define POW2(n) (1LL << (n)) #define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i=i##_begin_;i--) #define REP(i, n) FOR(i,0,n) #define IREP(i, n) IFOR(i,0,n) template void ndarray(vector &vec, int len) { vec.resize(len); } template void ndarray(vector &vec, int len, Args... args) { vec.resize(len); for (auto &v : vec) ndarray(v, args...); } template bool mmax(T &m, const T q) { if (m < q) {m = q; return true;} else return false; } template bool mmin(T &m, const T q) { if (m > q) {m = q; return true;} else return false; } template pair operator+(const pair &l, const pair &r) { return make_pair(l.first + r.first, l.second + r.second); } template pair operator-(const pair &l, const pair &r) { return make_pair(l.first - r.first, l.second - r.second); } template istream &operator>>(istream &is, vector &vec){ for (auto &v : vec) is >> v; return is; } ///// This part below is only for debug, not used ///// template ostream &operator<<(ostream &os, const vector &vec){ os << "["; for (auto v : vec) os << v << ","; os << "]"; return os; } template ostream &operator<<(ostream &os, const deque &vec){ os << "deq["; for (auto v : vec) os << v << ","; os << "]"; return os; } template ostream &operator<<(ostream &os, const set &vec){ os << "{"; for (auto v : vec) os << v << ","; os << "}"; return os; } template ostream &operator<<(ostream &os, const unordered_set &vec){ os << "{"; for (auto v : vec) os << v << ","; os << "}"; return os; } template ostream &operator<<(ostream &os, const multiset &vec){ os << "{"; for (auto v : vec) os << v << ","; os << "}"; return os; } template ostream &operator<<(ostream &os, const unordered_multiset &vec){ os << "{"; for (auto v : vec) os << v << ","; os << "}"; return os; } template ostream &operator<<(ostream &os, const pair &pa){ os << "(" << pa.first << "," << pa.second << ")"; return os; } template ostream &operator<<(ostream &os, const map &mp){ os << "{"; for (auto v : mp) os << v.first << "=>" << v.second << ","; os << "}"; return os; } template ostream &operator<<(ostream &os, const unordered_map &mp){ os << "{"; for (auto v : mp) os << v.first << "=>" << v.second << ","; os << "}"; return os; } #define dbg(x) cerr << #x << " = " << (x) << " (L" << __LINE__ << ") " << __FILE__ << endl; ///// END ///// /* #include #include #include using namespace __gnu_pbds; // find_by_order(), order_of_key() template using pbds_set = tree, rb_tree_tag, tree_order_statistics_node_update>; template using pbds_map = tree, rb_tree_tag, tree_order_statistics_node_update>; */ struct SizeAwareUnionFind { vector par, cou; SizeAwareUnionFind(int N) : par(N), cou(N) { REP(i, N) par[i] = i, cou[i] = 1; } int find(int x) { return (par[x] == x) ? x : (par[x] = find(par[x])); } void unite(int x, int y) { x = find(x), y = find(y); if (x == y) return; if (cou[x] < cou[y]) swap(x, y); par[y] = x, cou[x] += cou[y]; } int count(int x) { return cou[find(x)]; } bool same(int x, int y) { return find(x) == find(y); } }; vector> edge; vector color; vector uv; int main() { int N, K; cin >> N >> K; color.resize(N - 1); uv.resize(N - 1); edge.resize(N); REP(e, N - 1) { int u, v, c; cin >> u >> v >> c; u--, v--; uv[e] = make_pair(u, v); color[e] = c; edge[u].push_back(e); edge[v].push_back(e); } SizeAwareUnionFind uf(N - 1); REP(r, N) { map color2eid; for (auto eid : edge[r]) { int c = color[eid]; if (color2eid.count(c)) uf.unite(eid, color2eid[c]); else color2eid[c] = eid; } } lint ret = 0; // dbg(edge); REP(r, N) { map szz; for (auto eid : edge[r]) szz[uf.find(eid)] = uf.count(eid); lint sztot = 0; for (auto p : szz) sztot += p.second; for (auto p : szz) ret += (sztot - p.second) * p.second; // dbg(r); // dbg(szz); // lint sztot = accumulate(ALL(szz), 0LL); // for (auto v : szz) ret += (sztot - v) * v; } cout << ret / 2 << endl; }