问题:求不超过 k k k 的路径的数量有多少条。
将所有的路径分成两类。
- 两个点在同一个子树内。(递归处理每一个子树)
- 两个点不在同一个子树内。求出重心的每一个子树中所有的点到重心的距离,然后任选两个不同子树且两个距离的和不超过 k k k,求出他的选择方式数量即可。然而有可能两个点选出来的是同一个子树里的,这是一条不合法的路径。所以进行容斥原理,将在每一个子树里且路径的和不超过 k k k 的选择方式的数量全部从答案里减掉就是答案。
因此问题就转换成了:给你一些数,随便选两个数总和 ≤ k \le k ≤k 的方案数是多少。然后二分 / 双指针滑动(尺取法)即可。
// 重学代码风格
#include<vector>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e4+10;
vector<pair<int,int> >z[N];bool vis[N];
int pp[N],qq[N],n,m;
int sz(int u,int fa=-1){
if(vis[u])return 0;
int res=1;
for(int j=0;j<z[u].size();j++){
int i=z[u][j].first,w=z[u][j].second;if(i!=fa)res+=sz(i,u);}
return res;
}
int zx(int u,int fa,int tot,int&uu){
if(vis[u])return 0;
int sum=1,ms=0;
for(int j=0;j<z[u].size();j++){
int i=z[u][j].first,w=z[u][j].second;if(i!=fa){
int t=zx(i,u,tot,uu);
ms=max(ms,t),sum+=t;
}}ms=max(ms,tot-sum);
if(ms<=(tot>>1))uu=u;
return sum;
}
void gd(int u,int fa,int d,int&tt){
if(vis[u])return;qq[tt++]=d;
for(int j=0;j<z[u].size();j++){
int i=z[u][j].first,w=z[u][j].second;
if(i!=fa)gd(i,u,d+w,tt);
}
}
int g(int a[],int k){
sort(a,a+k);
int cnt=0;
int i=k-1,j=-1;
while(~i){
while(j+1<i&&a[j+1]+a[i]<=m)j++;
j=min(j,i-1);cnt+=j+1;
i--;
}return cnt;
}
void gb(int u,int&res){
int pt=0;
for(int j=0;j<z[u].size();j++){
int i=z[u][j].first,w=z[u][j].second;
int tt=0;
gd(i,-1,w,tt);
res-=g(qq,tt);
for(int k=0;k<tt;k++){
if(qq[k]<=m)res++;
pp[pt++]=qq[k];
}
}
res+=g(pp,pt);
}
int solve(int u){
if(vis[u])return 0;
int cnt=0;(void)zx(u,-1,sz(u),u);
vis[u]=true;
gb(u,cnt);
for(int j=0;j<z[u].size();j++){
int i=z[u][j].first,w=z[u][j].second;cnt+=solve(i);}
return cnt;
}
int main(){
ios_base::sync_with_stdio(false);cin.tie(0);
while(cin>>n>>m,n||m){
memset(vis,false,sizeof vis);
for(int i=0;i<=n;i++)z[i].clear();
for(int i=1;i<n;i++){
int a,b,c;cin>>a>>b>>c;
a--,b--;
z[a].push_back(make_pair(b,c));
z[b].push_back(make_pair(a,c));
}cout<<solve(0)<<'\n';
}
}