E、阔力梯的树

题：https://ac.nowcoder.com/acm/contest/4010/E?&headNav=acm

分析：dsu。贪心方法：考虑插入一个值x，对总体贡献，若查找在序列中左边有值  l ，则+加上(x-l)²，若右边有值r，则加上(r-x)²，若俩边都有值，则要减去插入失去的(r-l)²   ;

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
typedef long long ll;
const ll INF=0x3f3f3f3f;
set<int>s;
const int M=1e5+;
int sz[M],son[M],vis[M];
vector<int>g[M];
ll ans[M],tmp[M];
ll sum;
void dfs1(int u){
    sz[u]=;
    for(int i=;i<g[u].size();i++){
        int v=g[u][i];
        dfs1(v);
        sz[u]+=sz[v];
        if(sz[v]>sz[son[u]])
            son[u]=v;
    }
}
void update(int u,int k){
    if(k==-);
        s.erase(u);
    set<int>::iterator it=s.lower_bound(u);

    ll r=*it;
    it--;
    ll l=*it;
    if(k==)
        s.insert(u);
    if(l!=-INF&&r!=INF)///当插入/删除的值在中间
        sum-=(r-l)*(r-l)*k;

    if(l!=-INF)//当插入/删除的值左边有数
        sum+=(u-l)*(u-l)*k;

    if(r!=INF)///当插入/删除的值右边有数
        sum+=(r-u)*(r-u)*k;

    for(int i=;i<g[u].size();i++)
        if(!vis[g[u][i]])
            update(g[u][i],k);
}
void dfs2(int u,int sign){
    //cout<<u<<endl;
    ///
    for(int i=;i<g[u].size();i++){
        int v=g[u][i];
        if(v!=son[u])
            dfs2(v,);
    }
    if(son[u])
        dfs2(son[u],),vis[son[u]]=;
    update(u,);

    ans[u]=sum;
    vis[son[u]]=;
    if(!sign)
        update(u,-);
}
int main(){
    int n;
    s.insert(INF);
    s.insert(-INF);
    scanf("%d",&n);
    for(int u,i=;i<=n;i++){
        scanf("%d",&u);
        g[u].pb(i);
    }
    dfs1();
    dfs2(,);
    for(int i=;i<=n;i++)
        printf("%lld\n",ans[i]);
    return ;
}