Submission #882853

#	Time	Username	Problem	Language	Result	Execution time	Memory
882853		TomkeMonke	Power Plant (JOI20_power)	C++17	100 / 100	117 ms	29704 KiB

power

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 2e5 + 7;

vector<int> G[MAXN];
string s;

int path[MAXN], siz[MAXN];
int ans = 0;

void DFS(int v, int p){
	
	int maks = 0, sum = 0;
	
	for(auto u : G[v]){
		
		if(u == p) continue;
		DFS(u, v);
		sum += siz[u];
		maks = max(maks, siz[u]);
	}
	
	if(s[v] == '1'){
		
		path[v] = maks + 1;
		siz[v] = max(1, sum - 1);
	}
	
	else {
		
		path[v] = maks;
		siz[v] = sum;
	}
	
	ans = max(ans, max(path[v], siz[v]));
}





int main(){
	
	ios_base::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	
	int n;
	cin >> n;
	
	for(int i = 1; i < n; i++){
		
		int a, b;
		cin >> a >> b;
		
		G[a].push_back(b);
		G[b].push_back(a);
	}
	
	cin >> s;
	s = "#" + s;
	DFS(1, 1);
	
	cout << ans << '\n';
	
	return 0;
}

• Subtask 1: 6 / 6
• Subtask 2: 41 / 41
• Subtask 3: 53 / 53