Submission #3606044


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,int d){

  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,d+1);
  }

  return tmp;
}

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<<2*dfs(x,0)<<endl;


  return 0;
}

Submission Info

Submission Time
Task B - ツリーグラフ
User snow39
Language C++14 (GCC 5.4.1)
Score 0
Code Size 738 Byte
Status WA
Exec Time 1 ms
Memory 256 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 100
Status
AC × 2
AC × 7
WA × 13
Set Name Test Cases
Sample subtask0_sample_01.txt, subtask0_sample_02.txt
All subtask0_sample_01.txt, subtask0_sample_02.txt, subtask1_line01.txt, subtask1_line02.txt, subtask1_line03.txt, subtask1_line04.txt, subtask1_line05.txt, subtask1_line06.txt, subtask1_random01.txt, subtask1_random02.txt, subtask1_random03.txt, subtask1_random04.txt, subtask1_random05.txt, subtask1_random06.txt, subtask1_random07.txt, subtask1_random08.txt, subtask1_special01.txt, subtask1_special02.txt, subtask1_special03.txt, subtask1_special04.txt
Case Name Status Exec Time Memory
subtask0_sample_01.txt AC 1 ms 256 KB
subtask0_sample_02.txt AC 1 ms 256 KB
subtask1_line01.txt AC 1 ms 256 KB
subtask1_line02.txt WA 1 ms 256 KB
subtask1_line03.txt WA 1 ms 256 KB
subtask1_line04.txt WA 1 ms 256 KB
subtask1_line05.txt WA 1 ms 256 KB
subtask1_line06.txt AC 1 ms 256 KB
subtask1_random01.txt WA 1 ms 256 KB
subtask1_random02.txt WA 1 ms 256 KB
subtask1_random03.txt WA 1 ms 256 KB
subtask1_random04.txt WA 1 ms 256 KB
subtask1_random05.txt WA 1 ms 256 KB
subtask1_random06.txt WA 1 ms 256 KB
subtask1_random07.txt WA 1 ms 256 KB
subtask1_random08.txt WA 1 ms 256 KB
subtask1_special01.txt AC 1 ms 256 KB
subtask1_special02.txt AC 1 ms 256 KB
subtask1_special03.txt AC 1 ms 256 KB
subtask1_special04.txt WA 1 ms 256 KB