poj 1741 Tree(树分治)

第一次接触树分治，这个是点分治做的。
如果不了解树分治的话，可以先看看09年国家集训队的论文
https://wenku.baidu.com/view/1bc2e4ea172ded630b1cb602.html
https://wenku.baidu.com/view/60c6aa1ffc4ffe473368aba8.html
这两个一个pdf的，一个ppt。
参考：http://blog.csdn.net/qq_31759205/article/details/75095962
求dist(u,v)<=k有多少对，一条路径，要么过根节点，要么在子树内，所以递归处理。过根结点的路径还要分为路径的两头在同一子树内还是不同子树内。

处理前先求出树的重心，以重心为根节点来dfs，防止树退化成链。
先处理过根节点的路径，depth[i]表示节点到树根的距离。dfs求出来每个节点到树根的距离，升序排序，放俩标记i,j，一个指头，一个指尾，让尾部的标记往前移动，直到depth[i]+depth[j]<=k,这时距离根节点depth[i]的那个节点，就有j-i个节点和他相连的路径长度<=k了，然后i向右移动一下，继续重复上述步骤，条件是i < j。这就算出过根节点的路径了，路径两端在同一子树内的减掉就行。我也不大会表达，一看代码就懂了。
至于在子树内的路径，那就不过根节点了，递归处理就好了。

做的时候tle了一大堆，看了讨论版才发现，递归处理子树的时候，节点个数还是按照整个树的所有节点来的，这样就找不到重心了。

#include <stdio.h>#include <string.h>#include <algorithm>using namespace std;const int MAXN = 10010;const int MAXM = 20020;struct Edge{    int to,w,next;}edge[MAXM];int head[MAXN];int son[MAXN];int depth[MAXN];int dr;int n,k,tot,tmd;bool vis[MAXN];void init(){    tot = 0;    memset(son,0,sizeof(son));    memset(vis,0,sizeof(vis));    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++;}void getRoot(int N, int u, int fa, int &root){    son[u] = 1;    int ans = 0;    for(int i = head[u]; i != -1; i = edge[i].next)    {        int v = edge[i].to;        if(v == fa || vis[v]) continue;        getRoot(N,v,u,root);        son[u] += son[v];        ans = max(ans,son[v]);    }    ans = max(ans,N-son[u]);    if(ans < tmd)    {        tmd = ans;        root = u;    }}void getDepth(int u, int fa, int dep){    depth[dr++] = dep;    for(int i = head[u]; i != -1; i = edge[i].next)    {        int v = edge[i].to;        int w = edge[i].w;        if(v == fa || vis[v]) continue;        getDepth(v,u,dep+w);    }}int calc(int u, int dep){    dr = 0;    int ret = 0;    getDepth(u,-1,dep);    sort(depth,depth+dr);    int i = 0, j = dr-1;    while(i < j)    {                            //忘写i<j，wa了两发        while(depth[i]+depth[j] > k && i < j) --j;        ret += (j-i);        ++i;    }    return ret;}int solve(int u, int N){    int root,ret = 0;    tmd = MAXN;    getRoot(N,u,-1,root);    vis[root] = true;    ret += calc(root,0);    for(int i = head[root]; i != -1; i = edge[i].next)    {        int v = edge[i].to;        if(vis[v]) continue;        int w = edge[i].w;        ret -= calc(v,w);        ret += solve(v,son[v]);    }    return ret;}int main(){    int a,b,w;    while(scanf("%d %d",&n,&k) && n+k)    {        init();        for(int i = 0; i < n-1; ++i)        {            scanf("%d %d %d",&a,&b,&w);            addedge(a,b,w);            addedge(b,a,w);        }        printf("%d\n",solve(1,n));    }    return 0;}