poj1741 tree 点分治

把这几天写的东西上传一下
写博客的意义大概就是整理一下思路 以及方便日后复习
不得不说 把代码放在文件夹里
做完我是不会过目的
写博客 或许会增加我的成就感
从而 会多注意一下吧
哈哈 csdn现在也高大上了很多嘛
poj1741 Tree
题目大意:求(u,v)满足dis(u,v)<=k的点对数 N<=10000
点分治 LNWC2016新学到的 Orz
大概是 把所有的边分为经过点 x (重心) 和不经过点x
不经过点x的 多棵子树递归处理
可以证明递归次数不超过 logN 次
刚学完时候想尝试徒手写失败了= =
参考了hzwer的点分治= =
在这里%%%
但是我发现了个问题 点分治每次找的不是重心 但黄学长说没影响 不知道为什么
此题题解:
对于每次处理的根节点
求出它的子树所有点到距离 排序
ans+= dis(u,root)+dis(root,v)<=k 的点对数
但对于(u,v)在一棵子树上的情况 本不应该算入
所以答案还要减去(u,v)在一棵子树的情况

这个点分治是模仿hzwer实现的 有时间再写一发这题或别的题QAQ
在这里立个flag呦

#include<iostream>
#include<cstdio> 
#include<algorithm>
using namespace std;
const int N=10010;
const int inf=0x7fffffff;
struct E {int to,nxt,val;}edge[N*2];
int idx[N],can[N],size[N],deep[N],d[N],f[N];
int tot,root,ans,ret,sum,p;
int n,k;
void init(){
    tot=1;root=0;ans=0;sum=n;
    fill(idx,idx+N,0);
    fill(can,can+N,1);
    fill(size,size+N,0);
    fill(deep,deep+N,0);
    fill(d,d+N,0);
    fill(f,f+N,0);f[0]=inf;
}
void addedge(int from,int to,int val){
    edge[tot].to=to;edge[tot].val=val;edge[tot].nxt=idx[from];idx[from]=tot++;
}
void getroot(int x,int fa){
    size[x]=1;f[x]=0;
    for(int t=idx[x];t;t=edge[t].nxt){
        E e=edge[t];
        if(can[e.to] && e.to!=fa){
            getroot(e.to,x);
            f[x]=max(f[x],size[e.to]);
            size[x]+=size[e.to];
        }
    }
    f[x]=max(f[x],sum-size[x]);
    if(f[x]<f[root]) root=x;
}
void getdeep(int x,int fa){
    deep[++p]=d[x];
    for(int t=idx[x];t;t=edge[t].nxt){
        E e=edge[t];
        if(can[e.to] && e.to!=fa) {
            d[e.to]=d[x]+e.val;
            getdeep(e.to,x);
        }
    }
}
int calc(int x,int now){
    ret=0;d[x]=now;p=0;
    getdeep(x,0);
    sort(deep+1,deep+1+p);
    int r=p;
    for(int l=1;l<=p;l++){
        while(deep[l]+deep[r]>k) r--;
        if(r<=l) break;
        ret+=(r-l);
    }
    return ret;
}
void work(int x){
    ans+=calc(x,0);
    can[x]=0;
    for(int t=idx[x];t;t=edge[t].nxt){
        E e=edge[t];
        if(can[e.to]){
            sum=size[e.to];
            ans-=calc(e.to,e.val);
            root=0;
            getroot(e.to,0);
            work(root);
        }
    }
}
int main(){
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    while(1){
        scanf("%d%d",&n,&k);
        if(n==0 && k==0) break;
        init();
        for(int i=1;i<n;i++){
            int x,y,w;
            scanf("%d%d%d",&x,&y,&w);
            addedge(x,y,w);addedge(y,x,w);
        }
        getroot(1,0);
        work(root);
        printf("%d\n",ans);
    }
    return 0;
}