제출 #398062

#제출 시각아이디문제언어결과실행 시간메모리
398062prvocislo친구 (IOI14_friend)C++17
27 / 100
1 ms460 KiB
#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
typedef long long ll;
using namespace std;

const int maxn = 1e5 + 79;
int dp[maxn][2]; // dp[vr][isiel nas kamos?]
int dfs(int u, int fr, const vector<vector<pair<int, int> > >& g, int c[])
{
	if (dp[u][fr] != -1) return dp[u][fr];
	int go = c[u], dont = 0;
	for (const pair<int, int>& i : g[u])
	{
		int fr2 = (i.second ? fr : 0), p2 = (i.second != 1);
		go += dfs(i.first, (fr2 | p2), g, c);
		dont += dfs(i.first, fr2, g, c);
	}
	return dp[u][fr] = max(dont, go * (1 - fr));
}
int findSample(int n, int c[], int h[], int pr[])
{
	vector<vector<pair<int, int> > > g(n);
	for (int i = 1; i < n; i++)
		g[h[i]].push_back({ i, pr[i] });
	for (int i = 0; i < n; i++) for (int j = 0; j < 2; j++) dp[i][j] = -1;
	int ans = dfs(0, 0, g, c);
	//for (int i = 0; i < n; i++) for (int j = 0; j < 2; j++) cout << dp[i][j] << " \n"[j == 1];
	return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...