#include #include using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf 1000000000 struct Data{ array a; Data(){ rep(i,3)a[i] = Inf; } void add(int x){ if(x==Inf)return; rep(i,3){ if(a[i]==x)return; } rep(i,3){ if(a[i]==Inf){ a[i] = x; break; } } sort(a.begin(),a.end()); } }; template struct sparse_table{ vector v; int _n; int sz; sparse_table() : sparse_table(0) {} sparse_table(int n) : sparse_table(std::vector(n, e())) {} sparse_table(const std::vector& x) : _n(int(x.size())) { sz = x.size(); for(int i=0;true;i++){ if((1<_n){ sz = i; break; } } v.resize(sz*_n,e()); for(int i=0;i<_n;i++){ v[i] = x[i]; } for(int i=1;i> E; vector cnt; int r; vector pos; vector> bind; vector ind; vector> par; centroid_tree(vector> argE){ int n = argE.size(); E.resize(n,vector()); cnt.resize(n); r = go(0,argE); pos.resize(n); bind.resize(n); par.resize(n,make_pair(-1,-1)); dfs(r,-1,E); build(r,0); } int go(int a,vector> &argE){ dfs(a,-1,argE); int sz = cnt[a]; int ret = dfs2(a,-1,argE,sz); rep(i,argE[ret].size()){ int to = argE[ret][i]; argE[to].erase(find(argE[to].begin(),argE[to].end(),ret)); int g = go(to,argE); E[ret].push_back(g); } argE[ret].clear(); return ret; } void dfs(int cur,int p,vector> &argE){ cnt[cur] = 1; rep(i,argE[cur].size()){ int to = argE[cur][i]; if(to==p)continue; dfs(to,cur,argE); cnt[cur] += cnt[to]; } } int dfs2(int cur,int p,vector> &argE,int sz){ int ret = -1; int m = 0; rep(i,argE[cur].size()){ int to = argE[cur][i]; if(to==p)continue; ret = dfs2(to,cur,argE,sz); if(ret!=-1)return ret; m = max(m,cnt[to]); } if(sz-cnt[cur]<=sz/2 && m<=sz/2){ ret = cur; } return ret; } void build(int cur,int d){ int x = cur; pos[cur] = ind.size(); bind[cur].push_back(ind.size()); ind.push_back(cur); rep(i,E[cur].size()){ int to = E[cur][i]; int x = to; bind[cur].push_back(ind.size()); par[to] = make_pair(cur,i+1); build(to,d+1); } bind[cur].push_back(ind.size()); } }; struct HLD{ vector sz,parent,depth,root,pos; vector arr; HLD(vector> &E){ sz.resize(E.size(),1); parent.resize(E.size(),0); depth.resize(E.size(),0); root.resize(E.size(),0); pos.resize(E.size(),0); dfs(0,-1,E); dfs2(0,-1,E,0); } void dfs(int now,int p,vector> &E){ parent[now] = p; if(p==-1){ depth[now] = 0; } else{ depth[now] = depth[p]+1; } for(int i=0;i> &E,int r){ pos[now] = arr.size(); arr.push_back(now); root[now] = r; int maxi = 0; int ind = -1; for(int i=0;i> query(int u,int v){ vector> ret; int t = 0; while(root[u]!=root[v]){ if(depth[root[u]] <= depth[root[v]]){ ret.insert(ret.begin()+t,{pos[root[v]], pos[v]}); v = parent[root[v]]; } else{ ret.insert(ret.begin()+t,{pos[u],pos[root[u]]}); u = parent[root[u]]; t++; } } ret.insert(ret.begin()+t,{pos[u],pos[v]}); return ret; } int lca(int u,int v){ for(;;v=parent[root[v]]){ if(pos[u]>pos[v])swap(u,v); if(root[u]==root[v])return u; } } int get_distance(int u,int v){ return depth[u] + depth[v] - 2 * depth[lca(u,v)]; } }; Data op(Data a,Data b){ rep(i,3){ a.add(b.a[i]); } return a; } Data e(){ return Data(); } int main(){ int n,K; cin>>n>>K; vector D(2*n-1); vector> E(n); vector> E2(2*n-1); rep(i,n-1){ int u,v,c; scanf("%d %d %d",&u,&v,&c); u--;v--;c--; E[u].push_back(v); E[v].push_back(u); E2[u].push_back(n+i); E2[v].push_back(n+i); E2[n+i].push_back(u); E2[n+i].push_back(v); D[n+i].add(c); } HLD H(E2); { vector D2(2*n-1); rep(i,2*n-1){ D2[H.pos[i]] = D[i]; } swap(D,D2); } centroid_tree C(E); sparse_table seg(D); long long ans = 0LL; vector cnt1(K,0),cnt2(K,0); vector e1,e2; rep(i,n){ long long cnt1all = 0LL; map,long long> cntp; long long I = 0LL; rep(j,C.bind[i].size()-1){ vector V; for(int k=C.bind[i][j];kb)swap(a,b); t = op(t,seg.prod(a,b+1)); } V.push_back(t); } rep(k,V.size()){ if(V[k].a[0]==Inf)continue; if(V[k].a.back()!=Inf)continue; if(V[k].a[1]==Inf){ int x = V[k].a[0]; ans += cnt1all; ans -= cnt1[x]; ans += cnt2[x]; } else{ int x = V[k].a[0],y = V[k].a[1]; ans += I; ans += cnt1[x]; ans += cnt1[y]; if(cntp.count(make_pair(x,y)))ans += cntp[make_pair(x,y)]; } } rep(k,V.size()){ if(V[k].a[0]==Inf){ I++; continue; } if(V[k].a.back()!=Inf)continue; if(V[k].a[1]==Inf){ int x = V[k].a[0]; cnt1[x]++; e1.push_back(x); cnt1all++; } else{ int x = V[k].a[0],y = V[k].a[1]; cnt2[x]++; cnt2[y]++; e2.push_back(x); e2.push_back(y); cntp[make_pair(x,y)]++; } } } while(e1.size()>0){ cnt1[e1.back()]--; e1.pop_back(); } while(e2.size()>0){ cnt2[e2.back()]--; e2.pop_back(); } } cout<