Submission #1148944

---

Source Code Expand

#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <stack>
#include <queue>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <ctime>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;

static const double EPS = 1e-8;
static const double PI = 4.0 * atan(1.0);
static const double PI2 = 8.0 * atan(1.0);

#define REP(i,n)	for(int i=0;i<(int)n;++i)
#define ALL(c)		(c).begin(),(c).end()
#define CLEAR(v)	memset(v,0,sizeof(v))
#define MP(a,b)		make_pair((a),(b))
#define ABS(a)		((a)>0?(a):-(a))
#define FOR(i,s,n)	for(int i=s;i<(int)n;++i)

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

int res[100];

bool dfs(int v, int pa) {
	res[v] = 0;
	REP(i, es[v].size()) {
		if (es[v][i] == pa) continue;
		if (dfs(es[v][i], v)) res[v] += res[es[v][i]] + 2;
	}
	return res[v] != 0 || h[v] == 1;
}

int main(int argc, char **argv) {
	cin >> n >> x;
	--x;
	REP(i, n) cin >> h[i];
	REP(i, n - 1) {
		int a, b;
		cin >> a >> b;
		--a; --b;
		es[a].push_back(b);
		es[b].push_back(a);
	}

	dfs(x, -1);

	cout << res[x] << endl;

	return 0;
}

Submission Info

Judge Result