BZOJ 2103/3302/2447 消防站 树的重心【DFS】【TreeDP】

2103: Fire 消防站

Time Limit: 30 Sec  Memory Limit: 259 MB
Submit: 157  Solved: 116
[Submit][Status][Discuss]

Description

Input

共N+1行。 第一行有一个正整数N，表示区域的个数。 接下来有N-1行，每行两个整数u、v，表述区域u和区域v之间有一条道路。 最后一行有N个正整数，第i个正整数表示区域i的权值W(i)。

Output

包含一个正整数，为最小的S(x, y)的值。

Sample Input

5
1 2
1 3
3 4
3 5
5 7 6 5 4

Sample Output

14
【样例解释】
选取区域2和区域3。
【数据规模和约定】
用H表示距离区域1最远结点的距离，即d(1, u)的最大值。
对于30%的数据满足：2 ≤ N ≤ 5000、H ≤ 30
对于70%的数据满足：2 ≤ N ≤ 50000、H ≤ 30
对于100%的数据满足：2 ≤ N ≤ 50000、H ≤ 70、W(i) ≤ 100

 

http://blog.csdn.net/braketbn/article/details/51055715

 #pragma GCC optimize(2)
 #pragma G++ optimize(2)
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #include<cstdio>
 #include<cstring>

 #define ll long long
 #define N 50007
 #define inf 1000000007
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(!isdigit(ch)){if(ch=='-')f=-;ch=getchar();}
     while(isdigit(ch)){x=(x<<)+(x<<)+ch-'';ch=getchar();}
     return x*f;
 }

 int n,cut;
 int fa[N],dep[N],mx1[N],mx2[N];
 ll sum[N],res[N],ans;
 int cnt,hed[N],rea[N<<],nxt[N<<];

 void add(int u,int v)
 {
     nxt[++cnt]=hed[u];
     hed[u]=cnt;
     rea[cnt]=v;
 }
 void dfs(int u)
 {
     for (int i=hed[u];~i;i=nxt[i])
     {
         int v=rea[i];
         if(v==fa[u])continue;
         dep[v]=dep[u]+,fa[v]=u;
         dfs(v);
         sum[u]+=sum[v];
         res[u]+=res[v]+sum[v];
         if(!mx1[u]||sum[v]>sum[mx1[u]])mx2[u]=mx1[u],mx1[u]=v;
         else if(!mx2[u]||sum[v]>sum[mx2[u]])mx2[u]=v;
     }
 }
 void find_center(ll &ret,int rt,int x,ll k,int jd)
 {
     ret=min(ret,k);
     int v=mx1[x];
     if(v==cut||sum[mx2[x]]>sum[mx1[x]])v=mx2[x];
     if(!v)return;
     find_center(ret,rt,v,k+sum[rt]-*sum[v],jd);
 }
 void solve(int u)
 {
     for (int i=hed[u];~i;i=nxt[i])
     {
         int v=rea[i];cut=rea[i];
         if(v==fa[u])continue;
         ll gx=inf,gy=inf;
         for (int j=u;j;j=fa[j])sum[j]-=sum[cut];
         find_center(gx,,,res[]-res[cut]-dep[cut]*sum[cut],u);
         find_center(gy,cut,cut,res[cut],u);
         ans=min(ans,gx+gy);
         for (int j=u;j;j=fa[j])sum[j]+=sum[cut];
         solve(v);
     }
 }
 int main()
 {
     n=read();
     memset(hed,-,sizeof(hed));
     for (int i=;i<n;i++)
     {
         int x=read(),y=read();
         add(x,y),add(y,x);
     }
     for (int i=;i<=n;i++)
         sum[i]=read();
     ans=inf;
     dfs();
     solve();
     printf("%lld\n",ans);
 }