제출 #347393

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
347393		dolphingarlic	Power Plant (JOI20_power)	C++14	100 / 100	192 ms	27180 KiB

power

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;

string s;
vector<int> graph[200001];
int dp[200001], ans = 1;

void dfs(int node = 1, int parent = 0) {
	int mx = 0;
	for (int i : graph[node]) if (i != parent) {
		dfs(i, node);
		dp[node] += dp[i];
		mx = max(mx, dp[i]);
	}
	dp[node] = max(s[node - 1] - '0', dp[node] - s[node - 1] + '0');
	ans = max(ans, max(mx + s[node - 1] - '0', dp[node]));
}

int main() {
	cin.tie(0)->sync_with_stdio(0);
	int n;
	cin >> n;
	for (int i = 1; i < n; i++) {
		int a, b;
		cin >> a >> b;
		graph[a].push_back(b);
		graph[b].push_back(a);
	}
	cin >> s;
	dfs();
	cout << ans << '\n';
	return 0;
}

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