Submission #849769

#	Time	Username	Problem	Language	Result	Execution time	Memory
849769		Abito	Power Plant (JOI20_power)	C++14	0 / 100	0 ms	344 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){
    if (adj[node].size()>1) dp[node]=-a[node];
    else dp[node]=a[node];
    for (auto u:adj[node]) if (u!=p) dp[node]+=rec(u,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<<' '<<ansi<<endl;
    //for (int i=1;i<=n;i++) cout<<dp[i]<<' ';
    //cout<<endl;
    return 0;
}

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