【bzoj 1468】tree

传送门~

解题思路

点分治
代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<string>
#include<cmath>
#include<cstdlib>
using namespace std;
struct node{
    int x,y,num;
    node* ch[2];
}*root;
int hed[40005],nex[80005],lb[80005],cap[80005];
int s[40005],pa[40005];
int u,n,siz,inf,ans,lo,lc,lx;
void add(int x,int y,int num){
    lo++;
    nex[lo]=hed[x];
    hed[x]=lo;
    lb[lo]=y;
    cap[lo]=num;
}
int getsiz(int x,int fa){
    int rc=0;
    for(int i=hed[x];i!=0;i=nex[i])
    if(!pa[lb[i]] && lb[i]!=fa)
    rc+=getsiz(lb[i],x);
    return rc+1;
}
int getu(int x,int fa){
    int rc=0,flag=0;
    for(int i=hed[x];i!=0;i=nex[i])
    if(!pa[lb[i]] && lb[i]!=fa){
        int r0=getu(lb[i],x);
        if(r0*2>siz) flag=1;
        rc+=r0;
    }
    if((rc+1)*2<siz) flag=1;
    if(!flag) u=x;
    return rc+1;
}
void dfs(int x,int fa,int num){
    if(num>inf) return ;
    s[++lc]=num;
    for(int i=hed[x];i!=0;i=nex[i])
    if(!pa[lb[i]] && lb[i]!=fa)
    dfs(lb[i],x,num+cap[i]);
}
int ch(node* &p){
    if(p==NULL) return 0;
    if(p->y<=lx) return p->num;
    int mid=(p->x+p->y)>>1;
    if(mid>=lx) return ch(p->ch[0]);
    else return ch(p->ch[0])+ch(p->ch[1]);
}
void Insert(node* &p,int xx,int yy){
    if(p==NULL) {
        p=new node;p->num=0;
        p->x=xx;p->y=yy;
        p->ch[0]=p->ch[1]=NULL;
    }
    p->num++;
    if(xx==yy) return ;
    int mid=(xx+yy)>>1;
    if(lx<=mid) Insert(p->ch[0],xx,mid);
    else Insert(p->ch[1],mid+1,yy);
}
void Clear(node* &p){
    if(p==NULL) return ;
    Clear(p->ch[0]);
    Clear(p->ch[1]);
    delete(p);p=NULL;
}
void solve(int x){
    siz=getsiz(x,x);
    int tmp=getu(x,x);
    pa[u]=1;lc=0;
    for(int i=hed[u];i!=0;i=nex[i])
    if(!pa[lb[i]]){
        dfs(lb[i],u,cap[i]);ans+=lc;
        for(int i=1;i<=lc;i++){
            lx=inf-s[i];
            ans+=ch(root);
        }
        for(int i=1;i<=lc;i++){
            lx=s[i];
            Insert(root,1,inf);
        }
        lc=0;
    }
    Clear(root);
    for(int i=hed[u];i!=0;i=nex[i])
    if(!pa[lb[i]]) solve(lb[i]);
}
int main(){
    int ai,bi,ci;
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&ai,&bi,&ci);
        add(ai,bi,ci);add(bi,ai,ci); 
    }
    scanf("%d",&inf);
    solve(1);printf("%d",ans);
    return 0;
}