Submission #1453767

---

Source Code Expand

#include<iostream>
#include<vector>
using namespace std;

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

int dfs(int pos, int par){
	if(pos != x && v[pos].size() == 1){
		return (h[pos] ? 1 : 0);
	}else{
		int ret = 0;
		for(int u : v[pos]){
			if(u != par){
				int add = dfs(u, pos);
				ret += (add ? add + 1 : 0);
			}
		}
		return ret;
	}
}

int main(){
	cin >> n >> x;
	--x;
	for(int i = 0; i < n; ++i) cin >> h[i];
	for(int i = 0; i < n -1; ++i){
		int a, b; cin >> a >> b;
		--a; --b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	if(n == 1){
		cout << 0 << endl;
	}else{
		cout << dfs(x, -1) << endl;
	}
	return 0;
}

Submission Info

Judge Result