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[1007];int ins[1007],num[1007];
int cnt,head[1007],Next[1000007],rea[1000007];double val[1000007];

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=1;i<=n+m;i++)
        dis[i]=inf,ins[i]=0,num[i]=0;
    queue<int>q;
    q.push(1);dis[1]=0,ins[1]=num[1]=1;    
    while(!q.empty())
    {
        int u=q.front();q.pop();
        for (int i=head[u];i!=-1;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]=1;
                    q.push(v);
                    num[v]++;
                    if (num[v]>sqrt(n+m)) return false;
                }
            }
        }
        ins[u]=0;
    }
    return true;
}
int main()
{
    while(~scanf("%d%d%lf%lf",&n,&m,&l,&u))
    {
        cnt=0;double x;
        memset(head,-1,sizeof(head));
        for(int i=1;i<=n;i++)
            for(int j=1;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");
    }
}