// https://onlinejudge.u-aizu.ac.jp/beta/review.html#OUPC2023Day2/8737708 #include #include #include #include using namespace std; struct CentroidDecomposition{ // 重心分解後の木の深さ vector rank; // 部分木のサイズ vector sz; // 元のグラフ vector> g; // 重心分解後の木におけるグラフの親 vector par; // 重心分解後の木におけるグラフの子 (あまり使わないらしいが一応持つ) vector> child; int n; CentroidDecomposition(vector> &g) : n(g.size()), g(g), sz(g.size()), rank(g.size(), -1), par(g.size(), -1), child(g.size()) { _build(0,-1); } // 部分木のサイズを計算 void calc_size(int v, int p) { sz[v] = 1; for(int to : g[v]) { if(rank[to] == -1 && to != p) { calc_size(to, v); sz[v] += sz[to]; } } } void _build(int v, int p) { calc_size(v, -1); int tot = sz[v]; bool ok = false; int pp = -1; // いま見ている部分木での重心を見つける while(!ok) { ok = true; for(int to : g[v]) { if(rank[to] == -1 && to != pp && 2 * sz[to] > tot) { pp = v, v = to, ok = false; break; } } } par[v] = p; if(p != -1) { child[p].push_back(v); rank[v] = rank[p] + 1; } else { rank[v] = 0; } // 深さ優先でたどる for(int to : g[v]) { if(rank[to] == -1) { _build(to, v); } } } }; typedef long long ll; int main(){ int i,n,k; cin >> n >> k; vector> g(n); vector>> G(n); for(i=0;i> u >> v >> c; u--; v--; g[u].push_back(v); g[v].push_back(u); G[u].push_back({v,c}); G[v].push_back({u,c}); } CentroidDecomposition cd(g); // for(i=0;i> vec; vector,ll>> vec2; // bool deb = i==0 && false; auto dfs = [&](auto &&self,int v,int p,unordered_map &mp,int id) -> void { // if(deb){ // cout << "now = " << v << " " << mp.size() << " " << ans << "\n"; // for(auto p:mp) cout << p.first << " " << p.second << "\n"; // } if(mp.size()>=3) return; // i->v if(mp.size()==2){ ans++; auto it = mp.begin(); int c1 = (*it).first; it++; int c2 = (*it).first; if(c1>c2) swap(c1,c2); vec2[id][{c1,c2}]++; } if(mp.size()==1){ vec[id][(*mp.begin()).first]++; } for(auto e:G[v]){ if(e.first==p || cd.rank[e.first] mp1; mp1[e.second]++; vec.resize(id + 1); vec2.resize(id + 1); dfs(dfs,e.first,i,mp1,id); id++; } // cout << "i = " << i + 1 << " after_dfs " << ans << "\n"; // u->i->v; ll val1 = 0,val2 = 0,val3 = 0,val4 = 0; unordered_map mp2; int de = 0; for(auto mp:vec){ ll sum3 = 0; de++; for(auto p:mp){ // cout << "de = " << de << " " << p.first << " " << p.second << "\n"; mp2[p.first] += p.second; val1 += p.second; sum3 += p.second; val4 += p.second*p.second; } val3 += sum3*sum3; } for(auto p:mp2) val2 += p.second*p.second; val1 *= val1; ans += (val1 - val2 - val3 + val4)/2; // cout << "i = " << i + 1 << " after_calc " << ans << "\n"; // cout << val1 << " " << val2 << " " << val3 << " " << val4 << " " << vec.size() << "\n"; map,ll> mp3; for(int j=0;j