#include // created [2020/02/28] 21:45:15 #pragma GCC diagnostic ignored "-Wsign-compare" #pragma GCC diagnostic ignored "-Wsign-conversion" using i32 = int32_t; using i64 = int64_t; using u32 = uint32_t; using u64 = uint64_t; using uint = unsigned int; using usize = std::size_t; using ll = long long; using ull = unsigned long long; using ld = long double; template using arr = T (&)[n]; template using c_arr = const T (&)[n]; template using max_heap = std::priority_queue; template using min_heap = std::priority_queue, std::greater>; template constexpr T popcount(const T u) { return u ? static_cast(__builtin_popcountll(static_cast(u))) : static_cast(0); } template constexpr T log2p1(const T u) { return u ? static_cast(64 - __builtin_clzll(static_cast(u))) : static_cast(0); } template constexpr T msbp1(const T u) { return log2p1(u); } template constexpr T lsbp1(const T u) { return __builtin_ffsll(u); } template constexpr T clog(const T u) { return u ? log2p1(u - 1) : static_cast(u); } template constexpr bool ispow2(const T u) { return u and (static_cast(u) & static_cast(u - 1)) == 0; } template constexpr T ceil2(const T u) { return static_cast(1) << clog(u); } template constexpr T floor2(const T u) { return u == 0 ? static_cast(0) : static_cast(1) << (log2p1(u) - 1); } template constexpr bool btest(const T mask, const usize ind) { return static_cast((static_cast(mask) >> ind) & static_cast(1)); } template void bset(T& mask, const usize ind) { mask |= (static_cast(1) << ind); } template void breset(T& mask, const usize ind) { mask &= ~(static_cast(1) << ind); } template void bflip(T& mask, const usize ind) { mask ^= (static_cast(1) << ind); } template void bset(T& mask, const usize ind, const bool b) { (b ? bset(mask, ind) : breset(mask, ind)); } template constexpr T bcut(const T mask, const usize ind) { return ind == 0 ? static_cast(0) : static_cast((static_cast(mask) << (64 - ind)) >> (64 - ind)); } template bool chmin(T& a, const T& b) { return (a > b ? a = b, true : false); } template bool chmax(T& a, const T& b) { return (a < b ? a = b, true : false); } constexpr unsigned int mod = 1000000007; template constexpr T inf_v = std::numeric_limits::max() / 4; template constexpr Real pi_v = Real{3.141592653589793238462643383279502884}; auto mfp = [](auto&& f) { return [=](auto&&... args) { return f(f, std::forward(args)...); }; }; template T in() { T v; return std::cin >> v, v; } template T in_v(typename std::enable_if<(i == n), c_arr>::type) { return in(); } template auto in_v(typename std::enable_if<(i < n), c_arr>::type& szs) { const usize s = (usize)szs[i]; std::vector(szs))> ans(s); for (usize j = 0; j < s; j++) { ans[j] = in_v(szs); } return ans; } template auto in_v(c_arr szs) { return in_v(szs); } template auto in_t() { return std::tuple...>{in()...}; } struct io_init { io_init() { std::cin.tie(nullptr), std::ios::sync_with_stdio(false); std::cout << std::fixed << std::setprecision(20); } void clear() { std::cin.tie(), std::ios::sync_with_stdio(true); } } io_setting; int out() { return 0; } template int out(const T& v) { return std::cout << v, 0; } template int out(const std::vector& v) { for (usize i = 0; i < v.size(); i++) { if (i > 0) { std::cout << ' '; } out(v[i]); } return 0; } template int out(const std::pair& v) { return out(v.first), std::cout << ' ', out(v.second), 0; } template int out(const T& v, const Args... args) { return out(v), std::cout << ' ', out(args...), 0; } template int outln(const Args... args) { return out(args...), std::cout << '\n', 0; } template int outel(const Args... args) { return out(args...), std::cout << std::endl, 0; } # define SHOW(...) static_cast(0) constexpr ull TEN(const usize n) { return n == 0 ? 1ULL : TEN(n - 1) * 10ULL; } template auto make_v(typename std::enable_if<(i == n), c_arr>::type, const T& v = T{}) { return v; } template auto make_v(typename std::enable_if<(i < n), c_arr>::type szs, const T& v = T{}) { const usize s = (usize)szs[i]; return std::vector(szs, v))>(s, make_v(szs, v)); } template auto make_v(c_arr szs, const T& t = T{}) { return make_v(szs, t); } template struct edge { using cost_type = Cost; usize u, v; Cost c; edge(const usize u, const usize v) : u{u}, v{v}, c{1} {} edge(const usize u, const usize v, const Cost& c) : u{u}, v{v}, c{c} {} operator usize() const { return v; } usize from() const { return u; } usize to() const { return v; } Cost cost() const { return c; } friend std::ostream& operator<<(std::ostream& os, const edge& e) { return os << e.u << "->" << e.v << ":" << e.c; } }; template class base_graph { public: base_graph(const usize n) : v{n}, es(n), res(n) {} void add_edge(const usize u, const usize v, const bool bi = false) { es[u].emplace_back(u, v), res[v].emplace_back(v, u); if (bi) { es[v].emplace_back(v, u), res[u].emplace_back(u, v); } } template void add_edge(const usize u, const usize v, const Cost& c, const bool bi = false) { es[u].emplace_back(u, v, c), res[v].emplace_back(v, u, c); if (bi) { es[v].emplace_back(v, u, c), res[u].emplace_back(u, v, c); } } std::vector& operator[](const usize u) { return es[u]; } const std::vector& operator[](const usize u) const { return es[u]; } std::vector& from(const usize u) { return es[u]; } const std::vector& from(const usize u) const { return es[u]; } std::vector& to(const usize v) { return res[v]; } const std::vector& to(const usize v) const { return res[v]; } usize size() const { return v; } friend std::ostream& operator<<(std::ostream& os, const base_graph& g) { for (usize i = 0; i < g.v; i++) { for (const auto& e : g.es[i]) { os << e << '\n'; } } return os; } private: usize v; std::vector> es, res; }; template using base_tree = base_graph; using graph = base_graph>; using tree = base_graph>; template using cost_graph = base_graph>; template using cost_tree = base_graph>; template> class centroid { public: centroid(const base_tree& tree) : cs(tree.size()) { const std::size_t sz = tree.size(); std::vector sub(sz, 1); std::vector used(sz, false); auto size = [&, sz](auto&& self, const std::size_t s, const std::size_t p) -> std::size_t { sub[s] = 1; for (const auto& e : tree[s]) { const std::size_t to = e.to(); if (to == p or used[to]) { continue; } sub[s] += self(self, to, s); } return sub[s]; }; auto search = [&, sz](auto&& self, const std::size_t s, const std::size_t p, const std::size_t tot) -> std::size_t { for (const auto& e : tree[s]) { const std::size_t to = e.to(); if (p == to or used[to]) { continue; } if (sub[to] * 2 > tot) { return self(self, to, s, tot); } } return s; }; auto build = [&, sz](auto&& self, const std::size_t s, const std::size_t pc) -> std::size_t { const std::size_t tot = size(size, s, sz), c = search(search, s, sz, tot); used[c] = true; if (pc != sz) { cs.add_edge(pc, c); } for (const auto& e : tree[c]) { const std::size_t to = e.to(); if (not used[to]) { self(self, to, c); } } return c; }; build(build, 0, sz); } const tree& centros() const { return cs; } private: tree cs; }; int main() { const auto N = in(); const auto K = in(); cost_graph g(N); for (int i = 0; i < N - 1; i++) { const auto u = in() - 1, v = in() - 1, c = in() - 1; g.add_edge(u, v, c, true); } const auto cs = centroid(g).centros(); int c = 0; for (; c < N; c++) { if (cs.to(c).empty()) { break; } } SHOW(cs); std::vector used(N, false); ll ans = 0; mfp([&](auto&& self, const int c) -> void { used[c] = true; std::map, int> total; // [A,B,C]->個数 std::vector, int>> subs; // 部分木ごとの [A,B,C]->個数 total[std::set{}]++; for (const auto& e : g[c]) { subs.push_back(std::map, int>{}); const int ncolor = e.cost(); const int ns = e.to(); if (used[ns]) { continue; } mfp([&](auto&& self, const int s, const int p, const std::set& cols) -> void { total[cols]++; subs.back()[cols]++; for (const auto& e : g[s]) { const int to = e.to(); const int color = e.cost(); if (to == p) { continue; } std::set ncols = cols; ncols.insert(color); if (ncols.size() > 2) { continue; } self(self, to, s, ncols); } })(ns, c, std::set{ncolor}); } SHOW(c); SHOW(total); SHOW(subs); ll sum = 0; std::map total_monos; // [A?]->個数 std::vector> sub_monos(subs.size()); // 部分木ごとの [A?]->個数 for (const auto& p : total) { const auto& st = p.first; for (const int c : st) { total_monos[c] += p.second; } } for (const int nc : cs[c]) { self(self, nc); } for (int i = 0; i < subs.size(); i++) { for (const auto& p : subs[i]) { const auto& st = p.first; for (const int c : st) { sub_monos[i][c] += p.second; } } } auto count = [&](const std::map, int>& mp, const std::map& mono) { ll zero = 0; ll one = 0; ll two = 0; for (const auto& p : mp) { const auto& st = p.first; const int cnt = p.second; if (st.size() == 0) { zero += cnt; } else if (st.size() == 1) { one += cnt; } else if (st.size() == 2) { two += cnt; } else { SHOW("HOGEHOGE"); } } ll ans = 0; for (const auto& p : mp) { const auto& st = p.first; const int cnt = p.second; if (st.size() == 0) { ans += two + one + zero; } else if (st.size() == 1) { ans += (mono.count(*st.begin()) ? mono.at(*st.begin()) : 0LL) - cnt; } else if (st.size() == 2) { ans += cnt; } else { SHOW("HOGEHOGE"); } for (const int c : st) { } } return ans; }; ans += count(total, total_monos); for (int i = 0; i < subs.size(); i++) { ans -= count(subs[i], sub_monos[i]); } })(c); outln(ans); return 0; }