Submission #1030880

#	Time	Username	Problem	Language	Result	Execution time	Memory
1030880		Unforgettablepl	Security Guard (JOI23_guard)	C++17	37 / 100	775 ms	1048576 KiB

guard

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

#define int long long



int32_t main(){
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	int n,m,q;
	cin >> n >> m >> q;
	vector<int> arr(n+1);
	for(int i=1;i<=n;i++)cin>>arr[i];
	vector<vector<int>> adj(n+1);
	for(int i=1;i<n;i++){
		int a,b;
		cin >> a >> b;
		adj[a].emplace_back(b);
		adj[b].emplace_back(a);
	}
	int ans = 0;
	int currmax = 0;
	function<void(int,int,int,int)> dfs = [&](int x,int p,int curr,int curroffset){
		ans+=arr[p];
		curr+=arr[p];
		curr-=arr[x];
		if(curr<0){
			curroffset-=curr;
			currmax=max(currmax,curroffset);
			curr = 0;
		}
		for(int&i:adj[x])if(i!=p)dfs(i,x,curr,curroffset);
	};
	dfs(1,0,arr[1],0);
	cout << ans+currmax << '\n';
}

• Subtask 1: 12 / 12
• Subtask 2: 13 / 13
• Subtask 3: 12 / 12
• Subtask 4: 0 / 13
• Subtask 5: 0 / 8
• Subtask 6: 0 / 18
• Subtask 7: 0 / 24