#ifdef t9unkubj #include"template.h" //#include"template_no_debug.h" #else #undef _GLIBCXX_DEBUG #pragma GCC optimize("O3") #define dbg(...) 199958 using namespace std; #include using uint=unsigned; using ll=long long; using ull=unsigned long long; using ld=long double; using pii=pair; using pll=pair; templateusing vc=vector; templateusing vvc=vc>; templateusing vvvc=vvc>; using vi=vc; using vvi=vc; using vvvi=vc; using vl=vc; using vvl=vc; using vvvl=vc; templateusing smpq=priority_queue,greater>; templateusing bipq=priority_queue; #define rep(i,n) for(ll i=0;i<(ll)(n);i++) #define REP(i,j,n) for(ll i=(j);i<(ll)(n);i++) #define DREP(i,n,m) for(ll i=(n);i>=(m);i--) #define drep(i,n) for(ll i=((n)-1);i>=0;i--) #define all(x) x.begin(),x.end() #define rall(x) x.rbegin(),x.rend() #define mp make_pair #define pb push_back #define eb emplace_back #define fi first #define se second #define is insert #define bg begin() #define ed end() void scan(int&a) { cin >> a; } void scan(ll&a) { cin >> a; } void scan(string&a) { cin >> a; } void scan(char&a) { cin >> a; } void scan(uint&a) { cin >> a; } void scan(ull&a) { cin >> a; } void scan(bool&a) { cin >> a; } void scan(ld&a){ cin>> a;} template void scan(vector&a) { for(auto&x:a) scan(x); } void read() {} template void read(Head&head, Tail&... tail) { scan(head); read(tail...); } #define INT(...) int __VA_ARGS__; read(__VA_ARGS__); #define LL(...) ll __VA_ARGS__; read(__VA_ARGS__); #define ULL(...) ull __VA_ARGS__; read(__VA_ARGS__); #define STR(...) string __VA_ARGS__; read(__VA_ARGS__); #define CHR(...) char __VA_ARGS__; read(__VA_ARGS__); #define DBL(...) double __VA_ARGS__; read(__VA_ARGS__); #define LD(...) ld __VA_ARGS__; read(__VA_ARGS__); #define VC(type, name, ...) vector name(__VA_ARGS__); read(name); #define VVC(type, name, size, ...) vector> name(size, vector(__VA_ARGS__)); read(name); void print(int a) { cout << a; } void print(ll a) { cout << a; } void print(string a) { cout << a; } void print(char a) { cout << a; } void print(uint a) { cout << a; } void print(bool a) { cout << a; } void print(ull a) { cout << a; } void print(double a) { cout << a; } void print(ld a){ cout<< a; } template void print(vectora) { for(int i=0;i<(int)a.size();i++){if(i)cout<<" ";print(a[i]);}cout< void PRT(T a) { print(a); cout < void PRT(Head head, Tail ... tail) { print(head); cout << " "; PRT(tail...); return; } template bool chmin(T &x, F y){ if(x>y){ x=y; return true; } return false; } template bool chmax(T &x, F y){ if(x int popcount(T n){ return __builtin_popcountll(n); } template T sum(vc&a){ return accumulate(all(a),T(0)); } template T max(vc&a){ return *max_element(all(a)); } template T min(vc&a){ return *min_element(all(a)); } template void unique(vc&a){ a.erase(unique(all(a)),a.end()); } vvi readgraph(int n,int m,int off = -1){ vvi g(n); rep(i, m){ int u,v; cin>>u>>v; u+=off,v+=off; g[u].push_back(v); g[v].push_back(u); } return g; } vvi readtree(int n,int off=-1){ return readgraph(n,n-1,off); } template vc presum(vc &a){ vc ret(a.size()+1); rep(i,a.size())ret[i+1]=ret[i]+a[i]; return ret; } template vc &operator+=(vc &a,F b){ for (auto&v:a)v += b; return a; } template vc &operator-=(vc&a,F b){ for (auto&v:a)v-=b; return a; } template vc &operator*=(vc&a,F b){ for (auto&v:a)v*=b; return a; } #endif double pass_time=0; ll ans=0; vcG[int(2e5)]; struct centroid_decomposition{ vectorrem,si; vector>g; centroid_decomposition(int n):rem(n),si(n),g(n){} void add_edge(int a,int b){ g[a].push_back(b); g[b].push_back(a); } int get_centroid(int v){ vectorver; auto dfs=[&](auto&dfs,int u,int v)->void{ ver.emplace_back(u); si[u]=1; for(auto x:g[u]){ if(rem[x]||x==v)continue; dfs(dfs,x,u); si[u]+=si[x]; } };dfs(dfs,v,-1); int now=(int)1e9; int now_idx=-1; for(auto x:ver){ if(si[x]*2>=(int)ver.size()){ if(now>si[x]){ now=si[x]; now_idx=x; } } } assert(now_idx!=-1); return now_idx; } void decomposition(int v){ int t=get_centroid(v); assert(!rem[t]); rem[t]=1; //二色ペア map,int>ma0; //二色ペア,一色の累積 mapma1; dbg(t); //一色 int sms=0; mapma2; for(auto [x,c]:G[t]){ if(rem[x])continue; vcvs; auto dfs=[&](auto&dfs,int u,int v,unordered_set&st)->void{ dbg(u); vi tt; for(auto&x:st)tt.pb(x); if(tt.size()==2&&tt[0]>tt[1])swap(tt[0],tt[1]); vs.pb(tt); if(tt.size()==2){ ans+=ma0[{tt[0],tt[1]}]; ans+=ma2[tt[0]]+ma2[tt[1]]; ++ans; }else{ ans+=sms-ma2[tt[0]]; ans+=ma1[tt[0]]; } for(auto [x,c]:G[u]){ if(!rem[x]&&x!=v){ auto nxt=st; nxt.insert(c); if(nxt.size()<=2){ dfs(dfs,x,u,nxt); } } } }; unordered_setst{c}; dfs(dfs,x,t,st); dbg(vs); for(auto&x:vs){ if(x.size()==1)ma2[x[0]]++,sms++; else{ ma0[{x[0],x[1]}]++; ma1[x[0]]++; ma1[x[1]]++; } } } dbg(ans); for(auto x:g[t]){ if(!rem[x])decomposition(x); } rem[t]=0; } }; void solve(){ INT(n,k); centroid_decomposition g(n); rep(i,n-1){ INT(u,v,c); --u,--v; g.add_edge(u,v); G[u].pb({v,c}); G[v].pb({u,c}); } g.decomposition(0); PRT(ans); } signed main(){ cin.tie(0)->sync_with_stdio(0); pass_time=clock(); int t=1; //cin>>t; while(t--)solve(); pass_time=clock()-pass_time; dbg(pass_time/CLOCKS_PER_SEC); }