【点分治】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<<],w[MAXN<<],first[MAXN],next[MAXN<<],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=;
     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,-);
     int sum=,maxv=;
     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=;i<En;++i)
       ans[dis[i].second]+=(dis2+En-lower_bound(dis2,dis2+En,dis[i]))*op;
 }
 void solve(int U)
 {
     calc_sizes(U,-);
     int s=calc_centroid(U,-,size[U]).second;
     centroid[s]=;
     for(int i=first[s];i;i=next[i])
       if(!centroid[v[i]])
         solve(v[i]);
     en3=;
     ds[en3]=make_pair(,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=; calc_dis(v[i],s,w[i]);
           for(int i=;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,-);
           memcpy(ds+en3,td,en2*sizeof(Point));
           memcpy(ds2+en3,td2,en2*sizeof(Point));
           en3+=en2;
         }
     calc_pairs(ds,ds2,en3,);
     centroid[s]=;
 }
 void init()
 {
     memset(ans,,(n+)*sizeof(int));
     memset(first,,(n+)*sizeof(int));
     en=;
 }
 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=;i<n;++i)
             {
               scanf("%d%d%d",&a,&b,&c);
               AddEdge(a,b,c);
               AddEdge(b,a,c);
             }
         for(int i=;i<=n;++i)
             {
               scanf("%d",&is[i]);
               if(!is[i]) p[i]=make_pair(INF,INF);
               else p[i]=make_pair(,i);
             }
         dfs2(,-);
         dfs1(,-);
         solve();
         int Ans=;
         for(int i=;i<=n;++i) if(!is[i]) Ans=max(Ans,ans[i]);
         printf("%d\n",Ans);
       }
     return ;
 }