922 2022. 01. 28 12:55:34 Valaki2 Tevefarm cpp14 Elfogadva 50/50 72ms 20928 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long
#define pb push_back
#define mp make_pair
#define fi first
#define se second

int n;
vector<int> camel;
vector<int> parent;
vector<vector<int> > children;

pair<int, vector<int> > dfs(int cur) {
    int sum = 0;
    vector<int> merged;
    for(int nei : children[cur]) {
        pair<int, vector<int> > cur_res = dfs(nei);
        sum += cur_res.fi;
        for(int x : cur_res.se) {
            merged.pb(x);
        }
    }
    if(camel[cur] > sum) {
        return mp(camel[cur], vector<int> (1, cur));
    } else {
        return mp(sum, merged);
    }
}

void solve() {
    cin >> n;
    camel.assign(1 + n, 0);
    parent.assign(1 + n, 0);
    children.assign(1 + n, vector<int> (0, 0));
    for(int i = 1; i <= n; i++) {
        cin >> camel[i];
    }
    for(int i = 2; i <= n; i++) {
        cin >> parent[i];
        children[parent[i]].pb(i);
    }
    pair<int, vector<int> > answer = dfs(1);
    cout << answer.fi << "\n";
    cout << answer.se.size() << "\n";
    for(int x : answer.se) {
        cout << x << " ";
    }
    cout << "\n";
}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    solve();
    return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 50/50
1 Elfogadva 0/0 2ms 1820 KiB
2 Elfogadva 0/0 2ms 2016 KiB
3 Elfogadva 4/4 1ms 1872 KiB
4 Elfogadva 4/4 1ms 1900 KiB
5 Elfogadva 4/4 1ms 1924 KiB
6 Elfogadva 4/4 2ms 1968 KiB
7 Elfogadva 4/4 30ms 8888 KiB
8 Elfogadva 6/6 37ms 11016 KiB
9 Elfogadva 6/6 48ms 13360 KiB
10 Elfogadva 6/6 56ms 15436 KiB
11 Elfogadva 6/6 64ms 17816 KiB
12 Elfogadva 6/6 72ms 20928 KiB