Submission #3606076

---

Source Code Expand

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <cmath>
#include <map>
#include <queue>
#include <iomanip>
#define MOD 1000000007
typedef long long ll;
using namespace std;

int n,x;
int h[110];
vector<int> v[110];
bool used[110];

int dfs(int now){

  //cout<<now<<" "<<d<<endl;
  used[now]=1;
  int tmp=0;
  //if(h[now]==1) tmp=d;
  for(int i=0;i<v[now].size();i++){
    int u=v[now][i];
    if(used[u]==0) tmp+=dfs(u);
  }

  if(tmp>0) return tmp+1;
  else if(h[now]==1) return 1;
  else return 0;
}

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

  cout<<max(0,2*(dfs(x)-1))<<endl;


  return 0;
}

Submission Info

Judge Result