9574 2024. 02. 23 11:17:51 csaron71 Tevefarm cpp17 Hibás válasz 27/50 101ms 17020 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];
	}
	
}

int main() {
	long long n;
	cin >> n;
	ertek.resize(n);
	vissza.assign(n, 10);
	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";

	for (int i=0; i<n; i++) {
		if (vissza[apa[i]]==-10 && vissza[i]==10) {
			veg.push_back(i+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 Elfogadva 0/0 3ms 1960 KiB
2 Hibás válasz 0/0 4ms 2408 KiB
3 Részben helyes 2/4 3ms 2492 KiB
4 Részben helyes 2/4 3ms 2608 KiB
5 Elfogadva 4/4 3ms 2844 KiB
6 Részben helyes 2/4 3ms 3220 KiB
7 Részben helyes 2/4 43ms 10172 KiB
8 Részben helyes 3/6 52ms 11544 KiB
9 Részben helyes 3/6 61ms 13328 KiB
10 Részben helyes 3/6 68ms 14404 KiB
11 Részben helyes 3/6 90ms 15908 KiB
12 Részben helyes 3/6 101ms 17020 KiB