POJ1741-树的分治&树的重心&树上的路径问题-Tree

https://vjudge.net/problem/POJ-1741
给定一个树，问你他们之间距离小于等于k的点对有多少个
1 可以理解为二分这个树，最后到只有一个顶点时结束，为了让其尽可能的快，所以取重心，
2 取到一个重心后，每个点到重心的距离，用一个On的双向搜索计算小于等于k的点对。但是可能会存在没有过这个点的点对，所以把所有距离都加上和 重心邻接的边，去掉那些不经过 当前重心的点。
3 分而治之。

画的太抽象了。。
里面是内部点对qwq， 外边是那些符合条件的点对，
如果想求内部点对， 就在该子树和重心的连接点，再求一次距离，然后都加上 子树连接点 和重心的边长。相当于计算距离重心的节点， 在这里再求一个点对，这时候的点对肯定都是 这个子树里面而实际上是不经过 重心的。（这个要好好理解，我也害怕我以后忘记了qwq）
还有，我t了好多次，都是因为我分治 的时候，还要把这个树当前子树作为 树，那么他的最大值就是 子树的大小（子树减去 dfs顺序下的子树，用来判断那个反向的子树。。）；

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
typedef long long ll;
using namespace std;
const int MAXN = 10010;
const int INF = 0x3f3f3f3f;
struct Edge
{
    int to,next,w;
}edge[MAXN*2];
int head[MAXN],tot;
void init()
{
    tot = 0;
    memset(head,-1,sizeof(head));
}
void addedge(int u,int v,int w)
{
    edge[tot].to = v; edge[tot].w = w;
    edge[tot].next = head[u];head[u] = tot++;
}
bool vis[MAXN];
int size[MAXN],dep[MAXN];
int le,ri;
int dfssize(int u,int pre)//求子树
{
    size[u] = 1;
    for(int i = head[u];i != -1;i = edge[i].next)
    {
        int v = edge[i].to;
        if(v == pre || vis[v])continue;
        size[u] += dfssize(v,u);
    }
    return size[u];
}
int minn;
int root;
//找重心
void getroot(int u,int pre,int totnum)
{
    int maxx = totnum - size[u];//反向子树，注意totsum要随时改
    for(int i = head[u];i != -1;i = edge[i].next)
    {
        int v = edge[i].to;
        if(v == pre || vis[v])continue;
        getroot(v,u,totnum);
        maxx = max(maxx,size[v]);//
    }//得到的maxx是当前节点的最大子树、
    if(maxx < minn){minn = maxx; root = u;}
}
void dfsdepth(int u,int pre,int d)
{
    dep[ri++] = d;
    for(int i = head[u];i != -1;i = edge[i].next)
    {
        int v = edge[i].to;
        if(v == pre || vis[v])continue;
        dfsdepth(v,u,d+edge[i].w);//记录距离当前重心的长度 
    }
}
int n,k;
int getdep()
{    sort(dep,dep+ri);
    int r=ri-1;int l=0;
    int  ret=0;
    while(l<r){
        while(dep[l]+dep[r]>k&&l<r) r--;
        ret+=1ll*(r-l);
        l++;
    }
   return ret;//双向搜索  记录 点对。
}
ll ans;
void  solve(int u)
{   dfssize(u,-1);
//for(int i=1;i<=n;i++)
    //cout<<size[i]<<"i"<<endl;
    minn=1e8;
    ri=0;
    getroot(u,-1,size[u]);//m个点。
    dfsdepth(root,-1,0);
    //cout<<minn<<"genjiedian"<<root<<endl;
    ans+=1ll*getdep();
    //cout<<"ff"<<ans<<endl;
    vis[root]=true;
    for(int i=head[root];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(!vis[v]){
            ri=0;
            dfsdepth(v,0,edge[i].w);
            ans-=1ll*getdep();//减去内部的、
            //cout<<ans<<endl;
            solve(v);
        }
    }

    return ;
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int u,v,w;
    while(scanf("%d%d",&n,&k) == 2)
    {
        if(n == 0 && k == 0)break;
        init();
        for(int i = 1;i < n;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
            addedge(v,u,w);
        }
        memset(vis,false,sizeof(vis));
        ans=0;
        solve(1);
        printf("%lld\n",ans);
    }
    return 0;
}