BZOJ 2103/3302/2447 消防站

三倍经验 好像很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;
}