Submission #849771

#	Time	Username	Problem	Language	Result	Execution time	Memory
849771		Abito	Power Plant (JOI20_power)	C++14	47 / 100	50 ms	752 KiB

power

#include <bits/stdc++.h>
#define F first
#define S second
#define pb push_back
#define ppb pop_back
#define ep insert
#define endl '\n'
#define elif else if
#define pow pwr
#define sqrt sqrtt
#define int long long
#define y1 YONE
typedef unsigned long long ull;
using namespace std;
const int N=2005;
int n,dp[N];
vector<int> adj[N];
bool a[N];
int rec(int node,int p){
    //int s=bool(p);
    dp[node]=0;
    for (auto u:adj[node]) if (u!=p) dp[node]+=rec(u,node);
    if (adj[node].size()<=1) dp[node]+=a[node];
    else dp[node]-=a[node];
    return dp[node]=max(dp[node],(int)a[node]);
}
int32_t main(){
    ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
    cin>>n;
    for (int i=1;i<n;i++){
        int x,y;cin>>x>>y;
        adj[x].pb(y);adj[y].pb(x);
    }for (int i=1;i<=n;i++){char c;cin>>c;a[i]=c-'0';}
    int ans=0,ansi=0;
    for (int i=1;i<=n;i++){
        int x=rec(i,0);
        if (x>ans){
            ans=x;
            ansi=i;
        }
    }cout<<ans<<endl;
    //for (int i=1;i<=n;i++) cout<<dp[i]<<' ';
    //cout<<endl;
    return 0;
}

Compilation message (stderr)

power.cpp: In function 'int32_t main()':
power.cpp:34:15: warning: variable 'ansi' set but not used [-Wunused-but-set-variable]
   34 |     int ans=0,ansi=0;
      |               ^~~~

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