【POJ 1741】Tree(树分治)

之前一段时间做的，但是因为懒所以没有写题解。有点忘了现在重新做一下，还是乖乖写题解吧~

树分治主要的在于要找重心。对于那些需要统计满足条件的路径条数的题，我们就可以想到树分治。

对于某一个点，我们可以将答案分成经过这个点的和不经过这个点的，那么经过这个点的所有答案可以快速统计出来，剩下的就只用递归到子树中去找就好了。

还是多做题吧~

下面是代码：

#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 10005;
struct edge{
    int to,next;
    int c;
}e[N << 1];
int n,k;
int head[N],sz;
int size[N],mx[N];
int nsize,root;
int res;
bool done[N];
vector<int> dis;
void init(){
    memset(head,-1,sizeof(head));
    memset(done,0,sizeof(done));
    sz = 0;
}
void addedge(int u,int v,int c){
    e[sz].to = v;
    e[sz].c = c;
    e[sz].next = head[u];
    head[u] = sz ++;
}
void getroot(int u,int f){
    size[u] = 1;mx[u] = 0;
    for(int i = head[u] ; i != -1 ; i = e[i].next){
        int v = e[i].to;
        if(v == f || done[v]) continue;
        getroot(v,u);
        size[u] += size[v];
        mx[u] = max(mx[u],size[v]);
    }
    mx[u] = max(mx[u],nsize - size[u]);
    if(mx[u] < mx[root]) root = u;
}
void getdis(int u,int f,int dist){
    dis.push_back(dist);
    for(int i = head[u] ; i != -1 ; i = e[i].next){
        int v = e[i].to;
        if(v == f || done[v]) continue;
        getdis(v,u,dist + e[i].c);
    }
}
int cal(int u,int c){
    dis.clear();
    getdis(u,0,c);
    sort(dis.begin(),dis.end());
    int res = 0;
    int l = 0,r = dis.size() - 1;
    while(l < r){
        if(dis[l] + dis[r] <= k) res += r - l++;
            else r --;
    }
    return res;
}
void deal(int u){
    res += cal(u,0);
    done[u] = 1;
    for(int i = head[u] ; i != -1 ; i = e[i].next){
        int v = e[i].to;
        if(done[v]) continue;
        res -= cal(v,e[i].c);
        mx[0] = nsize = size[v];
        getroot(v,root = 0);
        deal(root);
    }
}
void solve(){
    init();
    int u,v,c;
    for(int i = 1 ; i < n ; i ++){
        scanf("%d%d%d",&u,&v,&c);
        addedge(u,v,c);
        addedge(v,u,c);
    }
    res = 0;
    mx[0] = nsize = n;
    getroot(1,root = 0);
    deal(root);
    printf("%d\n",res);
}
int main()
{
    int u,v;
    while(~scanf("%d%d",&n,&k)){
        if(!n && !k) break;
        solve();
    }
    return 0;
}