bzoj 3365 [Usaco2004 Feb]Distance Statistics 路程统计（点分治，单调）

【题意】

求树上长度不超过k的点对数目。

【思路】

和 Tree 一样一样的。

就是最后统计的时候别忘把根加上。

【代码】

 #include<set>
 #include<cmath>
 #include<queue>
 #include<vector>
 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #define trav(u,i) for(int i=front[u];i;i=e[i].nxt)
 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 using namespace std;

 typedef long long ll;
 const int N = 2e5+; 

 ll read() {
     char c=getchar();
     ll f=,x=;
     while(!isdigit(c)) {
         if(c=='-') f=-; c=getchar();
     }
     while(isdigit(c))
         x=x*+c-'',c=getchar();
     return x*f;
 }

 struct Edge {
     int v,w,nxt;
 }e[N<<];
 int en=,front[N];
 void adde(int u,int v,int w)
 {
     e[++en]=(Edge){v,w,front[u]}; front[u]=en;
 }

 int n,m,K,ans;
 int rt,size,list[N],vis[N],dis[N],siz[N],f[N],l1,l2;

 bool cmp(const int& x,const int& y)
 {
     return dis[x]<dis[y];
 }
 void get_root(int u,int fa)
 {
     siz[u]=; f[u]=;
     trav(u,i) if(!vis[e[i].v]&&e[i].v!=fa){
         int v=e[i].v;
         get_root(v,u);
         siz[u]+=siz[v];
         if(f[u]<siz[v]) f[u]=siz[v];
     }
     f[u]=max(f[u],size-siz[u]);
     if(f[u]<f[rt]) rt=u;
 }
 void dfs(int u,int fa)
 {
     list[++l1]=u;
     trav(u,i)
     if(!vis[e[i].v]&&e[i].v!=fa) {
         int v=e[i].v;
         dis[v]=dis[u]+e[i].w;
         dfs(v,u);
     }
 }
 int get_ans(int l,int r)
 {
     sort(list+l,list+r+,cmp);
     int j=r,ans=;
     for(int i=l;i<=r;i++) {
         while(j>i&&dis[list[j]]+dis[list[i]]>K) j--;
         ans+=j-i; if(i==j) break;
     }
     return ans;
 }
 void solve(int u)
 {
     vis[u]=; l1=l2=;
     trav(u,i) if(!vis[e[i].v]) {
         int v=e[i].v;
         dis[v]=e[i].w;
         dfs(v,-);
         ans-=get_ans(l2+,l1);
         l2=l1;
     }
     list[++l1]=u; dis[u]=;
     ans+=get_ans(,l1);
     trav(u,i) if(!vis[e[i].v]) {
         rt=; get_root(e[i].v,-);
         size=siz[e[i].v];
         solve(rt);
     }
 }

 int main()
 {
     n=read(),m=read();
     int u,v,w; char s[];
     FOR(i,,m) {
         u=read(),v=read(),w=read();
         adde(u,v,w),adde(v,u,w);
         scanf("%s",s);
     }
     K=read();
     size=f[]=n;
     get_root(,-);
     solve(rt);
     printf("%d\n",ans);
     return ;
 }