142892025-01-10 11:49:27Huni136Tevefarmcpp17Elfogadva 50/50157ms14576 KiB
#include <bits/stdc++.h>
using namespace std;

long long ert[100001];
set<int> szomsz[100001];
int n;
//ifstream fin("be.txt");
set<int> ered;
int akt;

void kiszed(int x)
{
   if(szomsz[x].size()==0)
   {
       if(akt!=x)
       {
           ered.erase(x);
       }
   }
   else
   {
        if(akt!=x)
       {
           ered.erase(x);
       }
       for(auto it:szomsz[x])
       {
           kiszed(it);
       }



   }
}

long long bejar(int x)
{
    //cout<<x;
   if(szomsz[x].size()==0)
   {
       // cout<<x;
       //cout<<1;
       akt=x;
       ered.insert(x);
           kiszed(x);
       return ert[x];
   }
   else
   {
        //cout<<x;
       long long s=0;
       for(auto it:szomsz[x])
       {
           s+=bejar(it);
       }
       long long maxi=max(ert[x],s);
       if(maxi==ert[x])
       {
           ered.insert(x);
           akt=x;
           kiszed(x);
       }
       return maxi;
   }
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>ert[i];
    }
    int x;
    for(int i=2;i<=n;i++)
    {
        cin>>x;
        szomsz[x].insert(i);
    }
    cout<<bejar(1)<<endl;
    cout<<ered.size()<<endl;
    for(auto it:ered)
    {
        cout<<it<<" ";
    }
    /*for(int i=1;i<=n;i++)
    {
        cout<<szomsz[i].size();
        for(auto it:szomsz[i])
        {
            cout<<it<<" ";
        }
        cout<<endl;
    }*/
    return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base50/50
1Elfogadva0/04ms4916 KiB
2Elfogadva0/06ms5356 KiB
3Elfogadva4/46ms4916 KiB
4Elfogadva4/46ms4916 KiB
5Elfogadva4/46ms4920 KiB
6Elfogadva4/44ms4960 KiB
7Elfogadva4/467ms9524 KiB
8Elfogadva6/675ms10348 KiB
9Elfogadva6/689ms11252 KiB
10Elfogadva6/6118ms12340 KiB
11Elfogadva6/6131ms13616 KiB
12Elfogadva6/6157ms14576 KiB