三倍经验 好像很6的样子 然并卵 我不是权限狗。。。(upd:划掉 嘿嘿嘿)
就是上一篇所提的第三题。
根据题意 可以用两点分开 两个联通块,所以枚举割哪条边
两个点则是两个联通块中的重心 至于为什么 自己好好想想吧~
注意要维护第一大和第二大子树,万一被割掉了呢 对不对。
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#define me(a,x) memset(a,x,sizeof a)
#define cp(a,x) memcpy(a,x,sizeof a)
using namespace std;
typedef long long LL;
const int N=50010;
struct node{int x,y,next;}a[2*N];
int n,first[N],len,size[N],s1[N],s2[N],dep[N],w[N],fa[N];
void ins(int x,int y)
{
a[++len].x=x,a[len].y=y,a[len].next=first[x],first[x]=len;
}
void dfs(int x,int f)
{
size[x]+=w[x];
for(int k=first[x];k;k=a[k].next)
{
int y=a[k].y;
if(y==f)continue;
dep[y]=dep[x]+1,fa[y]=x;
dfs(y,x); size[x]+=size[y];
if(size[s1[x]]<size[y])s2[x]=s1[x],s1[x]=y;
else if(size[s2[x]]<size[y])s2[x]=y;
}
}
int main()
{
int i,k,x,y;
scanf("%d",&n); len=0;
for(i=1;i<n;i++){scanf("%d%d",&x,&y); ins(x,y),ins(y,x);}
for(i=1;i<=n;i++)scanf("%d",&w[i]);
dfs(1,0); LL s=0;
for(i=2;i<=n;i++)s+=size[i];
LL mn=(1ll<<60);
for(k=1;k<=len;k+=2)
{
x=a[k].x,y=a[k].y;
if(dep[x]<dep[y])swap(x,y);
LL sum=s; int u,v;
for(u=fa[x];u;u=fa[u])size[u]-=size[x],sum-=size[x];
u=1;
while(1)
{
if(size[s1[u]]>size[s2[u]] && s1[u]!=x) v=s1[u];
else v=s2[u];
if(size[v]*2>size[1]) sum+=size[1]-2*size[v],u=v;
else break;
}
u=x;
while(1)
{
v=s1[u];
if(size[v]*2>size[x]) sum+=size[x]-2*size[v],u=v;
else break;
}
mn=min(mn,sum);
for(u=fa[x];u;u=fa[u])size[u]+=size[x];
}
printf("%lld\n",mn);
return 0;
}