POJ 1741 Tree【Tree,点分治】

给一棵边带权树，问两点之间的距离小于等于K的点对有多少个。

模板题：就是不断找树的重心，然后分开了，分治，至少分成两半，就是上限为log

然后一起统计就ok了

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<cmath>
 #include<queue>
 #include<algorithm>
 #define inf 2000000007
 using namespace std;

 int n,m;
 double l,u;
 double dis[];int ins[],num[];
 int cnt,head[],Next[],rea[];double val[];

 void add(int u,int v,double fee)
 {
     Next[++cnt]=head[u];
     head[u]=cnt;
     rea[cnt]=v;
     val[cnt]=fee;
 }
 bool Spfa()
 {
     for (int i=;i<=n+m;i++)
         dis[i]=inf,ins[i]=,num[i]=;
     queue<int>q;
     q.push();dis[]=,ins[]=num[]=;
     while(!q.empty())
     {
         int u=q.front();q.pop();
         for (int i=head[u];i!=-;i=Next[i])
         {
             int v=rea[i];double fee=val[u];
             if (dis[v]>dis[u]+fee)
             {
                 dis[v]=dis[u]+fee;
                 if (!ins[v])
                 {
                     ins[v]=;
                     q.push(v);
                     num[v]++;
                     if (num[v]>sqrt(n+m)) return false;
                 }
             }
         }
         ins[u]=;
     }
     return true;
 }
 int main()
 {
     while(~scanf("%d%d%lf%lf",&n,&m,&l,&u))
     {
         cnt=;double x;
         memset(head,-,sizeof(head));
         for(int i=;i<=n;i++)
             for(int j=;j<=m;j++)
             {
                 scanf("%lf",&x);
                 add(j+n,i,log(u)-log(x));
                 add(i,j+n,log(x)-log(l));
             }
         if(Spfa()) puts("YES");
         else puts("NO");
     }
 }