Submission #3207813

---

Source Code Expand

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <unordered_map>

using namespace std;
typedef long long int ll;
typedef pair<int, int> P;

int main()
{
	int n, x;
	cin>>n>>x;
  x--;
	vector<int> v;
	for(int i=0; i<n; i++){
		int h; cin>>h;
		if(h==1) v.push_back(i);
	}
	int d[100][100];
	for(int i=0; i<n; i++){
		for(int j=0; j<n; j++){
			if(i==j) d[i][j]=0;
			else d[i][j]=1e8;
		}
	}
  for(int i=0; i<n-1; i++){
    int a, b;
    cin>>a>>b;
    a--; b--;
    d[a][b]=1; d[b][a]=1;
  }
	for(int k=0; k<n; k++){
		for(int i=0; i<n; i++){
			for(int j=0; j<n; j++){
				d[i][j]=min(d[i][j], d[i][k]+d[k][j]);
			}
		}
	}
  int ans=0;
	for(auto y:v){
      ans+=d[x][y];
    }
  ans*=2;
	cout<<ans<<endl;
	return 0;
}

Submission Info

Judge Result