【点分治】hdu5016 Mart Master II

点分治好题。

①手动开栈。

②dp预处理每个点被哪个市场控制，及其距离是多少，记作pair<int,int>数组p。

③设dis[u].first为u到重心s的距离，dis[u].second=u，到在统计的时候，若dis[u]<=(p[v].first-dis[v].first,p[v].second)(双关键字比较)，则符合题意，当然这样在计算同一颗子树里的答案时，是压根不对的，但是既然这部分反正是要被减掉的，那就将错就错地计算即可了。

④最后在所有原非市场节点的ans中取一个最大的即可。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
using namespace std;
#define MAXN 100001
#define INF 1000000000
typedef pair<int,int> Point;
Point td[MAXN],ds[MAXN],p[MAXN],td2[MAXN],ds2[MAXN];
bool is[MAXN];
int n,K,ans[MAXN],en2,en3;
int v[MAXN<<1],w[MAXN<<1],first[MAXN],next[MAXN<<1],en;
void AddEdge(const int &U,const int &V,const int &W)
{
    v[++en]=V;
    w[en]=W;
    next[en]=first[U];
    first[U]=en;
}
bool centroid[MAXN];
int size[MAXN];
int calc_sizes(int U,int Fa)
{
    int res=1;
    for(int i=first[U];i;i=next[i])
      if(v[i]!=Fa&&(!centroid[v[i]]))
        res+=calc_sizes(v[i],U);
    return size[U]=res;
}
Point calc_centroid(int U,int Fa,int nn)
{
    Point res=make_pair(INF,-1);
    int sum=1,maxv=0;
    for(int i=first[U];i;i=next[i])
      if(v[i]!=Fa&&(!centroid[v[i]]))
        {
          res=min(res,calc_centroid(v[i],U,nn));
          maxv=max(maxv,size[v[i]]);
          sum+=size[v[i]];
        }
    maxv=max(maxv,nn-sum);
    res=min(res,make_pair(maxv,U));
    return res;
}
void calc_dis(int U,int Fa,int d)
{
    td[en2++]=make_pair(d,U);
    for(int i=first[U];i;i=next[i])
      if(v[i]!=Fa&&(!centroid[v[i]]))
        calc_dis(v[i],U,d+w[i]);
}
void calc_pairs(Point dis[],Point dis2[],int En,int op)
{
    sort(dis2,dis2+En);
    for(int i=0;i<En;++i)
      ans[dis[i].second]+=(dis2+En-lower_bound(dis2,dis2+En,dis[i]))*op;
}
void solve(int U)
{
    calc_sizes(U,-1);
    int s=calc_centroid(U,-1,size[U]).second;
    centroid[s]=1;
    for(int i=first[s];i;i=next[i])
      if(!centroid[v[i]])
        solve(v[i]);
    en3=0;
    ds[en3]=make_pair(0,s);
    ds2[en3++]=make_pair(p[s].first,p[s].second);
    for(int i=first[s];i;i=next[i])
      if(!centroid[v[i]])
        {
          en2=0; calc_dis(v[i],s,w[i]);
          for(int i=0;i<en2;++i)
            td2[i]=make_pair(p[td[i].second].first-td[i].first,p[td[i].second].second);
          calc_pairs(td,td2,en2,-1);
          memcpy(ds+en3,td,en2*sizeof(Point));
          memcpy(ds2+en3,td2,en2*sizeof(Point));
          en3+=en2;
        }
    calc_pairs(ds,ds2,en3,1);
    centroid[s]=0;
}
void init()
{
    memset(ans,0,(n+1)*sizeof(int));
    memset(first,0,(n+1)*sizeof(int));
    en=0;
}
void dfs1(int U,int Fa)
{
    for(int i=first[U];i;i=next[i])
      if(v[i]!=Fa)
        {
          p[v[i]]=min(p[v[i]],make_pair(p[U].first+w[i],p[U].second));
          dfs1(v[i],U);
        }
}
void dfs2(int U,int Fa)
{
    for(int i=first[U];i;i=next[i])
      if(v[i]!=Fa)
        {
          dfs2(v[i],U);
          p[U]=min(p[U],make_pair(p[v[i]].first+w[i],p[v[i]].second));
        }
}
int main()
{
    int a,b,c;
    while(scanf("%d",&n)!=EOF)
      {
          init();
        for(int i=1;i<n;++i)
            {
              scanf("%d%d%d",&a,&b,&c);
              AddEdge(a,b,c);
              AddEdge(b,a,c);
            }
        for(int i=1;i<=n;++i)
            {
              scanf("%d",&is[i]);
              if(!is[i]) p[i]=make_pair(INF,INF);
              else p[i]=make_pair(0,i);
            }
        dfs2(1,-1);
        dfs1(1,-1);
        solve(1);
        int Ans=0;
        for(int i=1;i<=n;++i) if(!is[i]) Ans=max(Ans,ans[i]);
        printf("%d
",Ans);
      }
    return 0;
}