poj1741 树上点分治

题意：给出树上的边权 询问所有i<j 且 i，j路径之和小于k的有多少对

解法：紫书上面引出的一个问题 解答大概是很长了 看了acmonster的题解才会做这里ORZ

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN=22222;
#define mp make_pair
struct node{
    int u,v,w,ne;
    node(int u=0,int v=0,int w=0,int ne=0):u(u),v(v),w(w),ne(ne){}
}e[MAXN];
int h[MAXN],num[MAXN],dep[MAXN],son[MAXN],sum[MAXN],vis[MAXN],f[MAXN];
int n,m,ans,l,r,i,a,b,c,root,sz,tp,len;
pair<int,int> q[MAXN];
void add(int a,int b,int c){
    node x;
    x.u=a;x.v=b;x.w=c;x.ne=h[a];
    e[sz]=x;
    h[a]=sz++;
}
void calc(int l,int r,int x){
	sort(num+l,num+r+1);
	for(int i=l,j=r;i<j;i++){
		while(i<j&&num[i]+num[j]>m)j--;
		ans+=x*(j-i);
	}
}
void dfs_0(int u,int fa){
	sum[u]=1;son[u]=0;
	for(int i=h[u];i!=-1;i=e[i].ne){
		int v=e[i].v;
		if(vis[v] && v!=fa){
			dfs_0(v,u);
			sum[u]+=sum[v];
			son[u]=max(son[u],sum[v]);
		}
	}
}
void dfs_1(int u,int fa){
	f[++len]=u;num[len]=dep[u];
	for(int i=h[u];i!=-1;i=e[i].ne){
		int v=e[i].v;
		if(vis[v]&&v!=fa){
			dep[v]=dep[u]+e[i].w;
			dfs_1(v,u);
		}
		if(u==root&&len>tp){
			if(tp+1<len)q[++r]=mp(tp+1,len);
			calc(tp+1,len,-1);
			tp=len;
		}
	}
}
void find(int l,int r){
	int tmp,sel;
	dfs_0(f[l],0);
	for(i=l,tmp=n+1;i <= r;i++){
		sel=max(son[f[i]],r-l-sum[f[i]]+1);
		if(sel<tmp)tmp=sel,root=f[i];
	}
}
void update(int l,int r){
	dep[root]=0;tp=l;len=l-1;
	dfs_1(root,0);
	for(i=l+1;i<=r;i++)	if(num[i]<=m)ans++;
	calc(l+1,r,1);
}
void work(int l,int r){
	if(l==r)return;
	for(i=l;i<=r;i++)vis[f[i]]=1;
	find(l,r);
	update(l,r);
	for(i=l;i<=r;i++)vis[f[i]]=0;
}
int main(){
    while(~scanf("%d%d",&n,&m)&&n){
        memset(h,-1,sizeof h);sz=0;
        for(int i=1;i<n;++i){
            scanf("%d%d%d",&a,&b,&c);add(a,b,c);add(b,a,c);
        }
        for(int i=1;i<=n;++i)f[i]=i;
        
		for(ans=0,q[l=r=1]=mp(1,n);l<=r;l++)
			work(q[l].first,q[l].second);
		printf("%d\n",ans);
	}
}