POJ 2114 Boatherds 点分治

   这道题是求等于k的，和求<=k的基本上是如出一辙，需要注意的是，我们在把子树距离排序以后，在寻找距离等于K的数目的时候，也是用两个指针，进行扫描，如果蛮子满足q[l]+q[r]==k，我们需要检查是不是这两个q[l]和q[r]是同一个值，如果是的话，我们其实就是在l 到 r中选取两个，用一下组合公式就知道是(r-l+1)*(r-l)/2，否则的话我们移动左指针算出a[l]个数 用右指针算出a[r]个数，然后相乘就是答案。

   需要注意的就是，开始我的理解是因为计算的时候，需要删除同一个联通块内的个数，然后递归到底部的时候，减去的满足a[l]+a[r]==k的不对，实际上是对的，等我想想怎么解释

  

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define LL long long
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxx = 5e5+6;
int ver[maxx],edge[maxx],head[maxx],Next[maxx];
int sz[maxx],vis[maxx];
LL q[maxx],d[maxx];
LL k,ans;
int n;
int root,tot,mx,size,l,r;
void add(int x,int y,int z){
   ver[++tot]=y;edge[tot]=z;Next[tot]=head[x];head[x]=tot;
   ver[++tot]=x;edge[tot]=z;Next[tot]=head[y];head[y]=tot;
}
void getroot(int u,int fa){
    sz[u]=1;
    int num=0;
    for (int i=head[u];i;i=Next[i]){
        int v=ver[i];
        if (fa==v||vis[v])continue;   ///注意vis和ver的区别
        getroot(v,u);
        sz[u]+=sz[v];
        num=max(num,sz[v]);
    }
    num=max(num,size-sz[u]);
    if(num<mx)mx=num,root=u;
}
void getdis(int u,int fa){
   q[++r]=d[u];
   for (int i=head[u];i;i=Next[i]){
      int v=ver[i];
      if(fa==v||vis[v])continue;
      d[v]=d[u]+edge[i];
      getdis(v,u);  ///TMD 写错了找了好久
   }
}
LL cal(int x,LL val){
    r=0;
    l=1;
    LL sum=0;
    d[x]=val;
    getdis(x,0);
    sort(q+1,q+1+r);
    while(l<r){
        if (q[l]+q[r]==k){
           if (q[l]==q[r]){
               sum+=(r-l+1)*(r-l)/2; ///注意是sum 不要写成ans
               break;
           }else{
               int i=l;
               int j=r;
               while(q[i]==q[l])i++;
               while(q[j]==q[r])j--;
               sum+=(i-l)*(r-j);   ///注意是sum 不要写成ans  
               l=i;
           }
        }else if (q[l]+q[r]<k)l++;
        else r--;
    }
    return sum;
}
void dfs(int u){
    vis[u]=1;
    ans+=cal(u,0);
    for (int i=head[u];i;i=Next[i]){
        int v=ver[i];
        if (vis[v])continue;
        ans-=cal(v,edge[i]);
        size=sz[v];
        mx=INF;
        getroot(v,0);
        dfs(root);
    }
}
void slove(){
   ans=0;
   size=n;  ///最开始size是为n的
   memset(vis,0,sizeof(vis));
   mx=INF;
   getroot(1,0);
   dfs(root);
}
int main(){
   int u,v,w;
   while(scanf("%d",&n)==1 && n){
      memset(head,0,sizeof(head));
      memset(Next,0,sizeof(Next));
      memset(sz,0,sizeof(sz));
      memset(d,0,sizeof(d));
      tot=0;
      for (int i=1;i<=n;i++){
        while(scanf("%d",&v) && v){
            scanf("%d",&w);
            add(i,v,w);
        }
      }
      while(scanf("%lld",&k)==1 && k){
         slove();
         if (ans)printf("AYE\n");
         else printf("NAY\n");
      }
      printf(".\n");
   }
  return 0;
}