Submission #955663

#	Time	Username	Problem	Language	Result	Execution time	Memory
955663		vjudge1	Power Plant (JOI20_power)	C++17	0 / 100	3 ms	9564 KiB

power

#include<bits/stdc++.h>
using namespace std;
int n;
int x,y;
vector<int>a[300000];
int ktr[300000];
int f[300000];
string s;
int asw=0;
void sol(int i){
    ktr[i]=1;
    int o=0;
    int mx=0;
    for(int u:a[i]){
        if(!ktr[u]){
            sol(u);
            f[i]+=f[u];
        }
    }
    if(s[i]=='1'){
        f[i]++;
        asw++;
        if(f[i]>2){
            asw--;
        }
    }
}
int main(){
    cin >>n;
    for(int i=1;i<n;i++){
        cin >>x>>y;
        a[x].push_back(y);
        a[y].push_back(x);
    }
    cin >>s;
    s=" "+s;
    sol(1);
    cout<<asw;
}
/*
5 4
2 1 3
1 3 5 6
1 1 2 4
2 2 3
*/

Compilation message (stderr)

power.cpp: In function 'void sol(int)':
power.cpp:12:9: warning: unused variable 'o' [-Wunused-variable]
   12 |     int o=0;
      |         ^
power.cpp:13:9: warning: unused variable 'mx' [-Wunused-variable]
   13 |     int mx=0;
      |         ^~

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