POJ 1741(点分治)

题目链接：http://poj.org/problem?id=1741

 

题目大意：一棵树上有多少条边小于k

 

题目思路：使用点分治，先建图，然后对树进行递归，先得到以1为根节点各个点的子节点数量，然后通过这个获取树的重心。树的重心就是所有点当中，他的子树的最大节点数最小。sum是当前树的所有节点数量，然后-siz[x]就是x这个节点面朝节点1的那棵子树的节点数量

然后就可以找出重心，接下来算出每个节点相对于根的距离，然后减去每一个字数中满足<k的边的个数，因为被重复计算了，容斥原理，然后再把这棵树所有满足的边都算出来，这时算出来的是跨根的满足<k的边的个数，不跨根的就直接判断跟根的距离是否小于k，因为是分治，所以同一个子树然后不经过根的小于k的边在之前已经算过了。

 

以下是代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
#define MAXN 10005
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
int n,k,tot,head[MAXN<<1],vis[MAXN],x,y,z,minn,siz[MAXN],l,r,dep[MAXN];
struct Edge{
    int to,next,w;
}edge[MAXN<<1];
void addedge(int u,int v,int w){
    edge[tot].to=v;edge[tot].w=w;edge[tot].next=head[u];head[u]=tot++;
}
int dfssize(int x,int pre){
    siz[x]=1;
    for(int i=head[x];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(vis[v]||v==pre)continue;
        siz[x]+=dfssize(v,x);
    }
    return siz[x];
}
void getroot(int x,int pre,int sum,int &root){
    int maxx=sum-siz[x];
    for(int i=head[x];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(vis[v]||v==pre)continue;
        getroot(v,x,sum,root);
        maxx=max(maxx,siz[v]);
    }
    if(maxx<minn){
        minn=maxx;
        root=x;
    }
}
void dfsdepth(int x,int pre,int w){
    dep[r++]=w;
    for(int i=head[x];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(vis[v]||pre==v)continue;
        dfsdepth(v,x,w+edge[i].w);
    }
}
int getdep(int l,int r){
    sort(dep+l,dep+r);
    int ret=0,rr=r-1;
    rep(i,l,r-1){
        if(dep[i]>k)break;
        while(rr>=l&&dep[i]+dep[rr]>k)rr--;
        ret+=rr-l+1;
        if(rr>i)ret--;
    }
    return ret>>1;
}
int solve(int x){
    minn=inf;
    int sum=dfssize(x,-1),root,ret=0;
    getroot(x,-1,sum,root);
    vis[root]=1;
    for(int i=head[root];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(vis[v])continue;
        ret+=solve(v);
    }
    l=r=0;
    for(int i=head[root];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(vis[v])continue;
        dfsdepth(v,root,edge[i].w);
        ret-=getdep(l,r);
        l=r;
    }
    ret+=getdep(0,r);
    rep(i,0,r-1){
        if(dep[i]<=k)ret++;
        else break;
    }
    vis[root]=0;
    return ret;
}
int main(){
    while(~scanf("%d%d",&n,&k)&&(n+k)){
        tot=0;
        memset(head,-1,sizeof(head));
        memset(vis,0,sizeof(vis));
        rep(i,1,n-1){
            scanf("%d%d%d",&x,&y,&z);
            addedge(x,y,z);
            addedge(y,x,z);
        }
        printf("%d\n",solve(1));
    }
    return 0;
}