【POJ1741】Tree 树分治 模板咯？

广告：

#include <stdio.h>int main(){    puts("转载请注明出处[vmurder]谢谢");    puts("网址：blog.csdn.net/vmurder/article/details/44302921");}

题意：

给你一棵无根树，求有多少点对之间距离<=K。

题解：

树分治。
然后对于一个重心X，我们把它的所有子树中的所有点存到结构体数组中。
结构体中存距离和子树编号。
第一遍sort，我们双指针扫哪些点对符合要求。
第二遍sort，我们把同一子树中还符合要求的点对数目删掉。
sort是O(nlogn)，处理是O(n)。
然后分治logn层，总时间复杂度O(nlog2n)

代码：

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#define N 50100#define inf 0x3f3f3f3fusing namespace std;int n,m;struct Eli{    int v,len,next;}e[N<<1];int head[N],cnt;bool rem[N];inline void add(int u,int v,int len){    e[++cnt].v=v;    e[cnt].len=len;    e[cnt].next=head[u];    head[u]=cnt;}int f[N],f2[N],son[N];int TreeCenter,length;int dfs1(int x,int p){    int i,v;    f[x]=f2[x]=0;    for(i=head[x];i;i=e[i].next)    {        v=e[i].v;        if(v==p||rem[v])continue;        int temp=dfs1(v,x);        if(temp>f[x])        {            f2[x]=f[x];            f[x]=temp;            son[x]=v;        }        else if(temp>f2[x])            f2[x]=temp;    }    return f[x]+1;}int g[N];void dfs2(int x,int p){    int i,v;    if(max(f[x],g[x])<length)    {        length=max(f[x],g[x]);        TreeCenter=x;    }    for(i=head[x];i;i=e[i].next)    {        v=e[i].v;        if(v==p||rem[v])continue;        if(v==son[x])g[v]=max(g[x],f2[x])+1;        else g[v]=max(g[x],f[x])+1;        dfs2(v,x);    }    return ;}inline int get_TreeCenter(int x){    if(rem[x])return 0;    length=inf,g[x]=0;    dfs1(x,0),dfs2(x,0);    return TreeCenter;}struct Summorer{    int a,b;    // 权值，子树标号    Summorer(int _a=0,int _b=0):a(_a),b(_b){}}src[N];inline int cmpa(Summorer a,Summorer b){    return a.a<b.a;}inline int cmpb(Summorer a,Summorer b){    return a.b==b.b?a.a<b.a:a.b<b.b;}int size[N];void dfs3(int x,int p,int t,int len){    src[++cnt]=Summorer(len,t);    size[x]=1;    int i,v;    for(i=head[x];i;i=e[i].next)    {        v=e[i].v;        if(rem[v]||v==p)continue;        dfs3(v,x,t,len+e[i].len);        size[x]+=size[v];    }    return ;}int ans;void work(int x,int n){    if(!n)return ;    rem[x=get_TreeCenter(x)]=1;    int num=0,i,j,k,v,p;    cnt=0;    for(i=head[x];i;i=e[i].next)    {        v=e[i].v;        if(rem[v])continue;        dfs3(v,x,++num,e[i].len);    }    sort(src+1,src+cnt+1,cmpa);    p=n;    for(i=0;i<=n;i++)    {        while(i<p&&src[i].a+src[p].a>m)p--;        if(i>=p)break;        ans+=(p-i);    }    sort(src+1,src+cnt+1,cmpb);    src[cnt+1].b=-1;    for(i=0;i<=n;i=j)    {        for(j=i+1;src[j].b==src[i].b;j++);        for(p=j-1,k=i;k<j;k++)        {            while(k<p&&src[k].a+src[p].a>m)p--;            if(k>=p)break;            ans-=(p-k);        }    }    for(i=head[x];i;i=e[i].next)    {        v=e[i].v;        if(rem[v])continue;        work(v,size[v]-1);    }    return ;}void init(){    ans=cnt=0;    memset(rem,0,sizeof rem);    memset(head,0,sizeof head);}int main(){    int i,a,b,c;    while(scanf("%d%d",&n,&m),(n|m))    {        init();        for(i=1;i<n;i++)        {            scanf("%d%d%d",&a,&b,&c);            add(a,b,c),add(b,a,c);        }        work(1,n-1);        printf("%d\n",ans);    }    return 0;}