9577 2024. 02. 23 11:40:02 csaron71 Tevefarm cpp17 Hibás válasz 27/50 100ms 16124 KiB
#include <bits/stdc++.h>
using namespace std;

vector<long long> ertek;
vector<long long> apa;
vector<vector<long long> > gyerekek;
vector<int> vissza;
vector<long long> veg;

long long kereses(long long x) {
	if (gyerekek[x].size()>0) {
		long long epp=ertek[x];
		long long masik=0;
		for (long long sz : gyerekek[x]) {
			masik+=kereses(sz);
		}
		if (epp>=masik) {
			//veg.push_back(x);
			vissza[x]=10;
			return epp;
		}
		else {
			vissza[x]=-10;
			return masik;
		}
		//return max(epp, masik);
	}
	else {
		//veg.push_back(x);
		//vissza[x]=-10;
		return ertek[x];
	}
	
}

void visszaf (int x) {
	if (vissza[x]==-10) {
		for (int sz : gyerekek[x]) {
			visszaf(sz);
		}
	}
	else {
		veg.push_back(x+1);
	}
}

int main() {
	long long n;
	cin >> n;
	ertek.resize(n);
	vissza.assign(n, 0);
	for (long long i=0; i<n; i++) {
		cin >> ertek[i];
	}
	apa.resize(n);
	apa[0]=0;
	gyerekek.assign(n, vector<long long>());
	for (long long i=1; i<n; i++) {
		cin >> apa[i];
		apa[i]--;
		gyerekek[apa[i]].push_back(i);
	}

	cout << kereses(0) << "\n";

	visszaf(1);

	//sort(veg.begin(), veg.end());
	cout << veg.size() << "\n";
	for (long long sz : veg) {
		cout << sz << " ";
	}
	cout << "\n";
	/*for (auto sz : vissza) {
		cout << sz << " ";
	}*/
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 27/50
1 Hibás válasz 0/0 3ms 1688 KiB
2 Hibás válasz 0/0 3ms 2044 KiB
3 Részben helyes 2/4 3ms 2060 KiB
4 Elfogadva 4/4 3ms 2284 KiB
5 Részben helyes 2/4 3ms 2508 KiB
6 Részben helyes 2/4 3ms 2880 KiB
7 Részben helyes 2/4 41ms 9360 KiB
8 Részben helyes 3/6 50ms 11020 KiB
9 Részben helyes 3/6 57ms 11896 KiB
10 Részben helyes 3/6 67ms 13672 KiB
11 Részben helyes 3/6 86ms 15076 KiB
12 Részben helyes 3/6 100ms 16124 KiB