Submission #1305220

#	Time	Username	Problem	Language	Result	Execution time	Memory
1305220		neonglitch	Power Plant (JOI20_power)	C++20	47 / 100	1596 ms	13072 KiB

power

#include <iostream>
#include <vector>
using namespace std;
const int N=2e5+10;
vector<int> ma[N];
int sp[N],dp[N];
void dfs(int x,int p=0)
{
	dp[x]=-sp[x];
	for(auto y:ma[x])
		if(y^p)
			dfs(y,x),dp[x]+=dp[y];
	dp[x]=max(dp[x],sp[x]);
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int n;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int x,y;
		cin>>x>>y;
		ma[x].push_back(y);
		ma[y].push_back(x);
	}
	for(int i=1;i<=n;i++)
	{
		char c;
		cin>>c;
		sp[i]=c-'0';
	}
	int ans=1;
	for(int i=1;i<=n;i++)
	{
		if(sp[i])
		{
			dfs(i);
			int sm=0;
			for(auto y:ma[i])
			{
				sm+=dp[y];
				ans=max(ans,dp[y]+1);
			}
			sm--;
			ans=max(ans,sm);
		}
	}
	cout<<ans<<endl;
}

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