小白月赛64D题 邻接表+dfs求以每个结点为根的子树大小

前言

码字不易 先赞后看 养成习惯

一、题目描述

在这里插入图片描述

二、问题分析

首先,我们是能够进行合法操作把任意两个a[i]的位置进行调换
在这里插入图片描述
在上述图中,比如如果我们想调换a[9]和a[2],那么我们可以把
2-1-6-8-9这条路上的权值序列翻转
在这里插入图片描述
显然,经过了反转,我们成功把a2和a9进行了对换
但是我们发现其中序号为1、6、8的结点也被改变了
但是根本不慌
我们发现再进行一次1-6-8这条路上的权值序列翻转
序号1、6、8即可变回原来的样子,即下图
在这里插入图片描述
观察上图任意一颗子树,比如以6为根的子树
我们发现结点9所对应的a2值最多在结点9结点1这条路上被乘一次
结点8所对应的a8值最多被乘2次,一次在结点9到结点1这条路上,另一次在结点8到结点1这条路上,其实也就是以结点8为根的子树大小 2

据此,我们所求值即为每个结点的子树大小该结点的a[i]值之和
∑ i = 1 n a [ i ] ∗ s z [ i ] \sum_{ i=1}^na[i]*sz[i] i=1na[i]sz[i]
其中sz[i]为每个结点的子树大小
那么怎么让这个值最大呢?
明显贪心的。让更大的a[i]乘更大的sz[i]就行
所以我们对a数组和sz数组sort排序,然后相乘再加起来就是答案。

三、邻接表+dfs求子树大小代码

void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u)
{
    int ans=1;
    st[u]=true;
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(!st[j]) ans+=dfs(j); 
    }
    sz[u]=ans;
    return ans;
}

四、最后,下面是题解代码(千万记得开longlong)

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int N =6e5+10;
int e[N],ne[N],h[N],w[N],idx;
int n;
int a[N];
bool st[N];
int sz[N];
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u)
{
    int ans=1;
    st[u]=true;
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(!st[j]) ans+=dfs(j); 
    }
    sz[u]=ans;
    return ans;
}
int main()
{
    memset(h,-1,sizeof h);
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=2;i<=n;i++)
    {
        int p;
        scanf("%d",&p);
        add(p,i);
        add(i,p);
    }
    int t=dfs(1);
    sort(a+1,a+1+n);
    sort(sz+1,sz+1+n);
    ll res=0;
    for(int i=1;i<=n;i++)
        res=res+(ll)a[i]*(ll)sz[i];
    cout<<res<<endl;
    return 0;
}
  • 9
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一支彩色铅笔

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值