95262024-02-22 14:46:02Vkrisztian01Tevefarmcpp17Elfogadva 50/50101ms15028 KiB
#include <iostream>
#include<vector>

using namespace std;


int n,a,b;
vector<vector<int> > g;
vector<long long int>label;
vector<long long int>dp;
vector<bool>valasztom;
vector<int>ki;

void dfs(int node)
{
    long long int osszeg=0;
    for(auto to:g[node])
    {
        dfs(to);
        osszeg+=dp[to];
    }
    dp[node]=max(label[node],osszeg);
    if(label[node]>osszeg) valasztom[node]=true;
    else valasztom[node]=false;
}

void bejar(int node)
{
    if(valasztom[node])
    {
        ki.push_back(node);
        return;
    }
    for(auto to:g[node]) bejar(to);
}

int main()
{
    cin>>n;
    g.resize(n+1);
    dp.assign(n+1,0);
    valasztom.resize(n+1);
    label.resize(n+1);
    for(int i=1;i<=n;i++) cin>>label[i];
    for(int i=2;i<=n;i++)
    {
        cin>>a;
        g[a].push_back(i);
    }
    dfs(1);
    cout<<dp[1]<<endl;
    bejar(1);
    cout<<ki.size()<<endl;
    for(auto x:ki) cout<<x<<" ";
    return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base50/50
1Elfogadva0/03ms1756 KiB
2Elfogadva0/03ms2096 KiB
3Elfogadva4/43ms2124 KiB
4Elfogadva4/43ms2380 KiB
5Elfogadva4/43ms2576 KiB
6Elfogadva4/43ms2948 KiB
7Elfogadva4/443ms8984 KiB
8Elfogadva6/652ms10124 KiB
9Elfogadva6/661ms11752 KiB
10Elfogadva6/668ms12780 KiB
11Elfogadva6/692ms13968 KiB
12Elfogadva6/6101ms15028 KiB