Poj 1741Tree（点分治+容斥)

http://poj.org/problem?id=1741

---

思路：
这个题不能树状数组水过去。因为有负权边。

所以对于去重的部分要用容斥去做。

对于一颗子树内的贡献，我们跑完了这个重心的子树后，对其子树再跑一遍去除贡献。此时消去贡献需要加重心到这个儿子节点的边权，加上一个偏移量。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e4+1000;
typedef int LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
struct Edge{LL to,w;Edge(LL _to,LL _w){to=_to;w=_w;}};
vector<Edge>g[maxn];
LL n;
long long k;
LL rt,siz[maxn],son[maxn],sum;
vector<long long> d;
bool vis[maxn];
void getroot(LL u,LL fa){
        siz[u]=1;son[u]=0;
        for(LL i=0;i<g[u].size();i++){
            LL v=g[u][i].to;
            if(v==fa||vis[v]) continue;
            getroot(v,u);
            siz[u]+=siz[v];
            if(siz[v]>son[u]) son[u]=siz[v];
        }
        son[u]=max(son[u],sum-siz[u]);
        if((son[u]<<1)<=sum) rt=u;
}
void getdis(LL u,LL fa,long long dis){
        d.push_back(dis);
        for(LL i=0;i<g[u].size();i++){
            LL v=g[u][i].to;long long w=g[u][i].w;
            if(v==fa||vis[v]) continue;
            getdis(v,u,dis+w);
        }
}
long long cal(LL u,LL fa,long long dep){///偏移量dep
        d.clear();
        getdis(u,fa,dep);
        LL cnt=d.size()-1;
        sort(d.begin(),d.end());
        LL i=0;LL j=cnt;long long res=0;
        while(i<j){
            while(i<j&&d[i]+d[j]>k) --j;
            res+=j-i;
            i++;
        }
        return res;
}
long long divide(LL u,LL fa){
        long long ans=0;
        vis[u]=true;
        ans+=cal(u,fa,0);
        for(LL i=0;i<g[u].size();i++){
            LL v=g[u][i].to;long long w=g[u][i].w;
            if(vis[v]||v==fa) continue;
            ans-=cal(v,u,w);
            son[rt=0]=sum=siz[v];
            getroot(v,0);
            getroot(rt,0);
            ans+=divide(rt,u);
        }
        return ans;
}
int main(void){
        while(1){
            n=read();k=read();
            if(n==0&&k==0) break;
            for(LL i=0;i<n+10;i++) g[i].clear(),vis[i]=0,siz[i]=0,son[i]=0,sum=0,rt=0;
            for(LL i=1;i<n;i++){
                LL u,v,w;u=read();v=read();w=read();
                g[u].push_back(Edge(v,w));g[v].push_back(Edge(u,w));
            }
            son[0]=sum=n;
            getroot(1,0);
            getroot(rt,0);
            long long ans=divide(rt,0);
            printf("%lld\n",ans);
       }
   return 0;
}