45522023-03-29 14:53:47AblablablaTevefarmcpp17Elfogadva 50/50104ms12760 KiB
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

int n;
vector<int> teve;
vector<vector<int>> graf;
vector<bool> jelolt;
vector<int> kivalasztott;

ll vizsgal(int helyzet){
    ll osszeg = 0;
    for(int kov : graf[helyzet]){
        osszeg += vizsgal(kov);
    }

    if(osszeg < teve[helyzet]){
        jelolt[helyzet] = true;
        osszeg = teve[helyzet];
    }

    return osszeg;
}

void dfs(int akt){
    if(jelolt[akt]){
        kivalasztott.push_back(akt);
        return;
    }
    for(int kov : graf[akt]){
        dfs(kov);
    }
}

int main()
{
    cin >> n;
    teve.assign(n, 0);
    graf.assign(n, vector<int>(0, 0));
    jelolt.assign(n, false);

    for(int i = 0; i < n; i++){
        cin >> teve[i];
    }

    for(int i = 0; i < n - 1; i++){
        int a;
        cin >> a;
        graf[a - 1].push_back(i + 1);
    }
/*
    for(int i = 0; i < n; i++){
        cout << i << ": ";
        for(int x : graf[i]){
            cout << x << " ";
        }

        cout << "\n";
    }
*/
    ll valasz = vizsgal(0);
    cout << valasz << "\n";
    dfs(0);
    cout << kivalasztott.size() << "\n";
    sort(kivalasztott.begin(), kivalasztott.end());
    for(int x : kivalasztott){
        cout << x+1 << " ";
    }
    cout << "\n";
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base50/50
1Elfogadva0/03ms1816 KiB
2Elfogadva0/03ms2180 KiB
3Elfogadva4/43ms2212 KiB
4Elfogadva4/43ms2468 KiB
5Elfogadva4/43ms2968 KiB
6Elfogadva4/43ms3100 KiB
7Elfogadva4/443ms7696 KiB
8Elfogadva6/654ms8800 KiB
9Elfogadva6/661ms9968 KiB
10Elfogadva6/671ms10896 KiB
11Elfogadva6/696ms11688 KiB
12Elfogadva6/6104ms12760 KiB