195062025-12-11 18:32:25KristófTevefarmcpp17Wrong answer 0/50123ms10660 KiB
#include <iostream>
#include <vector>
#include <algorithm>
#include <climits>
using namespace std;
vector<int> w;
vector<vector<int>> graph;
vector<vector<long long>> dp;
void solve(int s)
{
if(graph[s].size()==0)
    {
    dp[s][0]=0;
    dp[s][1]=w[s];
    return ;
    }
int sum=0;
dp[s][1]=w[s];
for(auto x : graph[s])
    {
    solve(x);
    dp[s][0]+=max(dp[x][0],dp[x][1]);
    //cout<<x<<" "<<dp[x][0]<<" "<<dp[x][1]<<endl;
    }
}

void backtrac(vector<int> &ans,int s)
{
if(dp[s][1]>=dp[s][0])
    {
    ans.push_back(s);
    return;
    }
for(auto x : graph[s])
    {
    backtrac(ans,x);
    }
}


int main()
{
    int n;
    cin>>n;
    w.resize(n+1);
    for(int i=1;i<=n;i++)cin>>w[i];
    //cout<<w[8]<<endl;
    graph.resize(n+1);
    dp.resize(n+1,vector<long long>(2));
    int x;
    for(int i=2;i<=n;i++)
        {
        cin>>x;
        graph[x].push_back(i);
        }
    solve(1);
    cout<<max(dp[1][0],dp[1][1]);
    vector<int> ans;
    backtrac(ans,1);
    cout<<ans.size()<<endl;
    for(auto x : ans)
    {

    cout<<x<<" ";
    }
    return 0;
}
SubtaskSumTestVerdictTimeMemory
base0/50
1Wrong answer0/01ms316 KiB
2Wrong answer0/02ms500 KiB
3Wrong answer0/41ms316 KiB
4Wrong answer0/41ms316 KiB
5Wrong answer0/41ms388 KiB
6Wrong answer0/42ms316 KiB
7Wrong answer0/448ms5476 KiB
8Wrong answer0/657ms6532 KiB
9Wrong answer0/672ms7600 KiB
10Wrong answer0/689ms8792 KiB
11Wrong answer0/6107ms9648 KiB
12Wrong answer0/6123ms10660 KiB