46872023-03-31 08:55:39Albertstein99Tevefarmcpp17Elfogadva 50/50103ms14700 KiB
#include <bits/stdc++.h>
using namespace std;

vector <vector <long long> > g;
vector <long long> W;
vector <bool> Q;
vector <long long> ans;


long long dfs(long long node){
    long long maxi = 0;
    for(auto x : g[node])
        maxi += dfs(x);
    if(maxi > W[node])
        return maxi;
    Q[node] = 1;
    return W[node];
}

void dfs2(long long node){
    if(Q[node]){
        ans.push_back(node);
        return;
    }
    for(auto x : g[node])
        dfs2(x);
}

int main(){
    long long N;
    cin >> N;
    W.resize(N + 1);
    g.resize(N + 1);
    Q.resize(N + 1);
    for(long long i = 1; i <= N; i++)
        cin >> W[i];
    for(long long i = 2; i <= N; i++){
        long long x;
        cin >> x;
        g[x].push_back(i);
    }
    cout << dfs(1) << '\n';
    dfs2(1);
    cout << ans.size() << '\n';
    for(auto x : ans){
        cout << x << ' ';
    }
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base50/50
1Elfogadva0/03ms1872 KiB
2Elfogadva0/03ms2312 KiB
3Elfogadva4/43ms2600 KiB
4Elfogadva4/42ms2564 KiB
5Elfogadva4/43ms2668 KiB
6Elfogadva4/43ms3036 KiB
7Elfogadva4/443ms8740 KiB
8Elfogadva6/652ms9980 KiB
9Elfogadva6/661ms11460 KiB
10Elfogadva6/668ms12740 KiB
11Elfogadva6/692ms13704 KiB
12Elfogadva6/6103ms14700 KiB