csp 202212-4聚集方差满分代码

MTS_ZX

已于 2023-03-17 21:15:07 修改

阅读量567

点赞数 1

分类专栏：刷题文章标签： c++ 算法

于 2023-03-17 15:18:17 首次发布

本文链接：https://blog.csdn.net/MTS_ZX/article/details/129621468

版权

刷题专栏收录该内容

4 篇文章 0 订阅

订阅专栏

该代码使用C++编写，实现了处理图数据结构并计算节点间距离的功能。它涉及到了深度优先搜索（DFS）来遍历图，以及在集合中进行排序和查找最近点的过程。程序首先读入图的边和节点值，然后通过两次DFS分别计算每个节点到子树中其他节点的距离。

摘要由CSDN通过智能技术生成

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
#include<set>
#include<vector>
using namespace std;
#define mp make_pair
#define ll long long
ll read()
{
    ll f=1,x=0;
    char c=getchar();
    while(c>'9'||c<'0')
    {
        if(c=='-') f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9')
    {
        x=x*10+c-'0';
        c=getchar();
    }
    return f*x;
}
#define N 300003
int n,head[N],Size,siz[N],a[N],son[N];
ll num[N];
ll as[N],nowAs;
struct setNode{
    int id;
    ll x;
    bool operator <(const setNode A)const
    {
        return x<A.x||x==A.x&&id<A.id;
    }
};
vector<int>ed[N];
set<setNode>se;
struct node{
    int next,to;
}mem[N<<1];
void add(int u,int v)
{
    mem[++Size].to=v;mem[Size].next=head[u];head[u]=Size;
}
void dfs1(int x,int fa)
{
    siz[x]=1;
    for(auto v:ed[x])
    {
        //int v=mem[i].to;
        if(v==fa) continue;
        dfs1(v,x);
        if(siz[v]>siz[son[x]]) son[x]=v;
        siz[x]+=siz[v];
    }
}
set<setNode>::iterator itl;
void ADD(setNode now)
{
    //cout<<nowAs<<'\n';
    if(se.empty())
    {
        num[now.id]=0;
        se.insert(now);
        return ;
    }
    if(se.size()==1)
    {
        itl=se.begin();
        ll t1=itl->x-now.x;
        num[itl->id]=num[now.id]=t1*t1;
        nowAs=t1*t1*2;
        se.insert(now);
        return ;
    }

    itl= se.upper_bound(now);
    ll tr=1e9,tl=1e9;
    if(itl!=se.end())
    {
        tr=itl->x-now.x;
        if(tr*tr<num[itl->id])
        {
            nowAs+=tr*tr-num[itl->id];
            num[itl->id]=tr*tr;
        }
    }
    if(itl!=se.begin())
    {
        --itl;
        tl=now.x-itl->x;
        if(tl*tl<num[itl->id])
        {
            nowAs+=tl*tl-num[itl->id];
            num[itl->id]=tl*tl;
        }
    }
    se.insert(now);
    num[now.id]=min(tl*tl,tr*tr);
    nowAs+=num[now.id];

}
void calc(int x,int fa)
{
    ADD(setNode{x,a[x]});
    for(auto v:ed[x])
    {
       // int v=mem[i].to;
        if(v==fa) continue;
        calc(v,x);
    }
}
void dfs2(int x,int fa)
{
    for(auto v:ed[x])
    {
        //int v=mem[i].to;
        if(v==fa||v==son[x]) continue;
        dfs2(v,x);
        se.clear();
        nowAs=0;
    }

    if(son[x]) dfs2(son[x],x);
    ADD(setNode{x,a[x]});
   // cout<<x<<" "<<son[x]<<" "<<nowAs<<" "<<se.size()<<'\n';
    for(auto v:ed[x])
    {
        //int v=mem[i].to;
        if(v==fa||v==son[x]) continue;
        calc(v,x);
    }
    as[x]=nowAs;
}
int main()
{

    n=read();
    for(int i=2;i<=n;++i)
    {
        int x=read();
        ed[x].push_back(i);
    }
    for(int i=1;i<=n;++i) a[i]=read();
    dfs1(1,0);
    dfs2(1,0);
    for(int i=1;i<=n;++i) cout<<as[i]<<'\n';
    return 0;
}