Submission #849772

#	Time	Username	Problem	Language	Result	Execution time	Memory
849772		Abito	Power Plant (JOI20_power)	C++14	0 / 100	2 ms	6492 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=2e5+5;
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';}
    cout<<rec(1,0);
    return 0;
}

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