Submission #1350893

---

Source Code Expand

#include <bits/stdc++.h>
 
using namespace std;
const int INF=1e9;

int n,x;
int h[105],cnt[105]; 
vector<int> v[105];

bool dfs(int x,int y){
    if(v[x].size()==1){
        if(h[x]){
            cnt[x]++; //cout<<x<<endl;
            return 1;
        }
        return 0;
    }
    for(int i=0; i<v[x].size(); i++){
        if(v[x][i]==y) continue;
        if(dfs(v[x][i],x)){
            if(!h[x]){
                cnt[x]++; //cout<<x<<endl;
            }
        }
    }
    if(h[x]){
        cnt[x]++; //cout<<x<<endl;
        return 1;
    }
    else return 0;
}

int main(){

cin>>n>>x; x--;

for(int i=0; i<n; i++) scanf("%d",&h[i]);

for(int i=0; i<n-1; i++){
    int a,b; scanf("%d %d",&a,&b); a--; b--;
    v[a].push_back(b);
    v[b].push_back(a);
}

v[x].push_back(-1); 

dfs(x,-1); 

int P=0;
for(int i=0; i<n; i++) if(cnt[i]) P++;

cout<<2*(P-1)<<endl;

}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:37:41: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
 for(int i=0; i<n; i++) scanf("%d",&h[i]);
                                         ^
./Main.cpp:40:34: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     int a,b; scanf("%d %d",&a,&b); a--; b--;
                                  ^

Judge Result