bzoj3924 幻想乡战略游戏

一开始，naive地以为只要重心向下找，然后暴力往上搞就行了，但是写加调1，2个小时wa了几遍后才发现这个找的做法是错误的。然后想了很久没收获，％了一发别人的程序，看了不久就看懂了。就是说我们知道第一次从根节点进入一个重心的时候一定是找到num值最大且满足大于总结点数一半的儿子，但是这个结论只对第一次成立！所以我们想办法将进入一次后，其他子树的信息结合到进入的子树中，然后记录下修改信息，最后反着做修改就可以复原。这样差不多等于把进入的重心看做一个新的根节点，因为我们已经把之前的信息合并到了子树中，所以那个结论我们又可以使用了！

/**************************************************************
    Problem: 3924
    User: Clare
    Language: C++
    Result: Accepted
    Time:28676 ms
    Memory:41828 kb
****************************************************************/
 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
 
#define N 100010
#define INF 0x3f3f3f3f
 
int n,Sum,root,Q,New_root;
struct Edge{
    int to,value,next;
}edge[N*2],edge1[N*2*10];
int head[N],head1[N],tot,tot1;
int size[N],f[N],dis[N];
bool done[N];
long long num[N],sum[N],S_num[N],S_sum[N],total,Ans;
struct Data{
    int to;
    long long Enter_pos,value;
};
vector g[N];
 
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
 
void Addedge1(int u,int v,int w)
{
    tot1++;edge1[tot1].next=head1[u];edge1[tot1].to=v;edge1[tot1].value=w;head1[u]=tot1;
}
 
void Addedge(int u,int v,int w)
{
    tot++;edge[tot].next=head[u];edge[tot].to=v;edge[tot].value=w;head[u]=tot;
    tot++;edge[tot].next=head[v];edge[tot].to=u;edge[tot].value=w;head[v]=tot;
}
 
void Get_root(int k,int fa)
{
    size[k]=1;f[k]=0;
    for(int i=head[k];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==fa||done[v])
            continue;
        Get_root(v,k);
        size[k]+=size[v];
        f[k]=max(f[k],size[v]);
    }
    f[k]=max(f[k],Sum-f[k]);
    if(f[k] Record;
void Query()
{
    int x=New_root,Max=0;
    Record.clear();
    while(x)
    {
        Max=0;
        for(int i=1;inum[g[x][Max].to])
                Max=i;
        if(g[x].size()==0||num[g[x][Max].to]*2<=num[x])
        {
            Ans=sum[x];
            for(int i=0;i