BZOJ3302: [Shoi2005]树的双中心

n<=50000的树，深度<=100，有点权，选两个点x，y，使最小。

dis取了min之后，整个树就会以某条边为分界线分成两半，一半归一个点管。如果是两棵完全独立的树的话，那肯定分别取这两棵树的带权重心。但割掉某条边再找两边重心，这种情况不一定是合法情况。例如：

上图中，虚线边被断开，两边的重心分别是星标节点。这不是一个合法方案，但它显然不如一个合法方案的答案优：

所以放心大胆地割就好了。注意到本题中树的深度h很小，所以割边后涉及的子树信息修改操作都可以暴力修改。

把树以某点为根，希望能预处理出一些信息使得能够在O(h)的时间内找到割完边每一半的重心。如果是普通的一棵树，知道了子树大小（权值和）之后，从某点出发最多走2h步之后就可以到重心。割了某条边之后，一个点最多只会有一棵子树信息被修改。因此记最大儿子和次大儿子即可。

至于答案的记录我写的有点丑。如果想的话可以看一下我怎么写的，不然就直接看怎么写更方便吧。所有子树大小加起来即为根节点为重心的总代价，因为这样加起来，越是下面的点加的次数越多。断掉某条边之后，修改子树大小的同时可修改根节点为重心答案，然后从根节点开始向重儿子走，只要比较最大儿子和次大儿子子树权和即可，边走边改答案。走一步，就是走到的这棵子树权值和少算一次，其外面的权值和多算一次。

我是记了每个节点为根的答案。。式子有点乱。。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
//#include<queue>
//#include<iostream>
using namespace std;
 
int t,n;
#define maxn 50011
struct Edge{int to,next;}edge[maxn<<1];int first[maxn],le,val[maxn];
void in(int x,int y) {Edge &e=edge[le];e.to=y;e.next=first[x];first[x]=le++;}
void insert(int x,int y) {in(x,y);in(y,x);}
void init()
{
    scanf("%d",&n);
    memset(first,0,sizeof(first));le=2;
    int x,y;
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        insert(x,y);
    }
    for (int i=1;i<=n;i++) scanf("%d",&val[i]);
}
#define LL long long
int size[maxn],dep[maxn],f[maxn],tot,hea[maxn],h2[maxn];LL aa[maxn];
void dfs(int x,int fa)
{
    size[x]=val[x];dep[x]=dep[fa]+1;f[x]=fa;tot+=val[x];
    hea[x]=h2[x]=0;aa[x]=0;
    for (int i=first[x];i;i=edge[i].next)
    {
        const Edge &e=edge[i];if (e.to==fa) continue;
        dfs(e.to,x);size[x]+=size[e.to];
        aa[x]+=aa[e.to]+size[e.to];
        if (size[e.to]>size[hea[x]]) h2[x]=hea[x],hea[x]=e.to;
        else if (size[e.to]>size[h2[x]]) h2[x]=e.to;
    }
}
int main()
{
        init();
        tot=0;dfs(1,0);
        LL ANS=1e15;
        for (int i=2;i<le;i+=2)
        {
            int x=edge[i].to,y=edge[i^1].to;
            if (dep[x]<dep[y]) {int t=x;x=y;y=t;}
            for (int i=f[x],cnt=1;i;i=f[i],cnt++) size[i]-=size[x],aa[i]-=aa[x]+1ll*cnt*size[x];
            tot-=size[x];
            LL ans=0,qq=0;
            for (int now=1;;)
            {
                int tmp=(hea[now]==x?(h2[now]?size[h2[now]]:0):(h2[now]==x?size[hea[now]]:
                max(size[hea[now]],size[h2[now]])))*2;
                 
                if (tmp<=tot) {ans+=aa[now]+qq;break;}
                if (size[hea[now]]>size[h2[now]] && hea[now]!=x) now=hea[now],
                qq+=aa[f[now]]-(aa[now]+size[now])+tot-size[now];
                else now=h2[now],qq+=aa[f[now]]-(aa[now]+size[now])+tot-size[now];
            }
            qq=0;
            for (int now=x;;)
            {
                if (size[hea[now]]*2<=size[x]) {ans+=aa[now]+qq;break;}
                now=hea[now],qq+=aa[f[now]]-(aa[now]+size[now])+size[x]-size[now];
            }
            ANS=min(ANS,ans);
            tot+=size[x];
            for (int i=f[x],cnt=1;i;i=f[i],cnt++) size[i]+=size[x],aa[i]+=aa[x]+1ll*cnt*size[x];
        }
        printf("%lld
",ANS);
    return 0;
}