男人八题之Tree / POJ1741

男人八题

问题：求不超过 $k$ 的路径的数量有多少条。

将所有的路径分成两类。

1. 两个点在同一个子树内。（递归处理每一个子树）
2. 两个点不在同一个子树内。求出重心的每一个子树中所有的点到重心的距离，然后任选两个不同子树且两个距离的和不超过 $k$，求出他的选择方式数量即可。然而有可能两个点选出来的是同一个子树里的，这是一条不合法的路径。所以进行容斥原理，将在每一个子树里且路径的和不超过 $k$ 的选择方式的数量全部从答案里减掉就是答案。

因此问题就转换成了：给你一些数，随便选两个数总和 $\le 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';
  }
}