洛谷 P4178 Tree

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define rg register
#define _ 40010
using namespace std;
int n,record[_],num_of_edges,size[_],dfn[_],dfns,F[_],root,dis[_];//size[]不多说,F[]是i点儿子中最大子树大小
bool passed[_];
int stack[_],top,ans,k;
struct pp
{
    int next,to,w;
}edge[_<<];
inline int read()
{
    rg int save=,w=;rg char q=getchar();
    while(q<''||q>''){if(q=='-')w=-;q=getchar();}
    while(q>=''&&q<='')save=(save<<)+(save<<)+q-'',q=getchar();
    return save*w;
}
void getroot(rg int i,rg int fa,rg int s)
{
    size[i]=,F[i]=,dfn[i]=++dfns;
    for(rg int j=record[i];j;j=edge[j].next)
    {
        rg int to=edge[j].to;
        if(to!=fa&&!passed[to])
        {
            getroot(to,i,s);
            F[i]=max(F[i],size[to]);
            size[i]+=size[to];
        }
    }
    F[i]=max(F[i],s-size[i]);
    root=F[i]<F[root]?i:root;
}
void comput(rg int i,rg int dad)
{
    stack[++top]=dis[i];//stack直接把边权记下来,无需记点
    for(rg int j=record[i];j;j=edge[j].next)
    {
        rg int to=edge[j].to;
        if((!passed[to])&&dad!=to)
        {
            dis[to]=dis[i]+edge[j].w;
            comput(to,i);
        }
    }
}
inline int doit(rg int i,rg int d)//数出 该联通块中若使所有路径强制经过重心,<=k的有多少    (d为初始距离(去重用))
{
    top=;dis[i]=d;
    comput(i,);//计算各点到该重心的路径长
    sort(stack+,stack+top+);
    rg int l=,r=top,result=;
    while(l<r)
        if(stack[l]+stack[r]<=k)result+=r-l,l++;
        else r--;
    return result;
}
void dfs(rg int i,rg int all)
{
    ans+=doit(i,);
    passed[i]=;
    for(rg int j=record[i];j;j=edge[j].next)
    {
        rg int to=edge[j].to;
        if(!passed[to])
        {
            rg int s;
/* ☆☆☆*/    ans-=doit(to,edge[j].w);//去掉那些 在to的子树加上了[i->to的边权]被算作i的贡献的路径(这些路径本不存在)
            if(dfn[i]<dfn[to])s=size[to];
            else s=all-size[i];
            root=;dfns=;
            getroot(to,i,s);
//            printf("%d\n",root);
            dfs(to,s);
        }
    }
}
inline void add(rg int from,rg int to,rg int ww)
{
    edge[++num_of_edges]=(pp){record[from],to,ww};
    record[from]=num_of_edges;
}
int main()
{
    n=read();
//    cout<<"___\n"<<n<<endl;
    rg int i,j;
    for(i=;i<n;++i)
    {
        rg int x=read(),y=read(),ww=read();
        add(x,y,ww),add(y,x,ww);
//        cout<<x<<' '<<y<<' '<<ww<<endl;
    }
    k=read();
//    cout<<k<<endl;
    root=;F[]=n;//操作 "root=F[i]<F[root]?i:root" 需要
    getroot(,,n);
    dfs(root,n);
    printf("%d\n",ans);
    return ;
}