Submission #286596
Source Code Expand
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cfloat>
#include <map>
#include <utility>
#include <set>
#include <iostream>
#include <memory>
#include <string>
#include <vector>
#include <algorithm>
#include <functional>
#include <sstream>
#include <deque>
#include <complex>
#include <stack>
#include <queue>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <ctime>
#include <iterator>
#include <bitset>
#include <numeric>
#include <list>
#include <iomanip>
#if __cplusplus >= 201103L
#include <array>
#include <tuple>
#include <initializer_list>
#include <unordered_set>
#include <unordered_map>
#include <forward_list>
#define cauto const auto&
#define ALL(v) begin(v),end(v)
#else
#define ALL(v) (v).begin(),(v).end()
#endif
using namespace std;
namespace{
typedef long long LL;
typedef pair<int,int> pii;
typedef pair<LL,LL> pll;
typedef vector<int> vint;
typedef vector<vector<int> > vvint;
typedef vector<long long> vll, vLL;
typedef vector<vector<long long> > vvll, vvLL;
#define VV(T) vector<vector< T > >
template <class T>
void initvv(vector<vector<T> > &v, int a, int b, const T &t = T()){
v.assign(a, vector<T>(b, t));
}
template <class F, class T>
void convert(const F &f, T &t){
stringstream ss;
ss << f;
ss >> t;
}
#define REP(i,n) for(int i=0;i<int(n);++i)
#define RALL(v) (v).rbegin(),(v).rend()
#define MOD 1000000007LL
#define EPS 1e-8
VV(int) G;
vint hs;
int dfs(int u, int prv){
int ret = 0;
for(int v : G[u]){
if(v == prv){ continue; }
ret += max(dfs(v, u) + 1, 0);
}
if(!hs[u] && !ret){
ret = -1010101010;
}
return ret;
}
void mainmain(){
int n, x;
cin >> n >> x;
--x;
hs.resize(n);
for(int i = 0; i < n; ++i){
cin >> hs[i];
}
G.resize(n);
for(int i = 0; i < n - 1; ++i){
int a, b;
cin >> a >> b;
G[a - 1].push_back(b - 1);
G[b - 1].push_back(a - 1);
}
int res = max(dfs(x, -1), 0) * 2;
printf("%d\n", res);
}
}
int main() try{
mainmain();
}
catch(...){}
Submission Info
Submission Time |
|
Task |
B - ツリーグラフ |
User |
climpet |
Language |
C++11 (GCC 4.8.1) |
Score |
100 |
Code Size |
2092 Byte |
Status |
AC |
Exec Time |
23 ms |
Memory |
932 KB |
Judge Result
Set Name |
Sample |
All |
Score / Max Score |
0 / 0 |
100 / 100 |
Status |
|
|
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 |
22 ms |
860 KB |
subtask0_sample_02.txt |
AC |
21 ms |
800 KB |
subtask1_line01.txt |
AC |
22 ms |
924 KB |
subtask1_line02.txt |
AC |
21 ms |
928 KB |
subtask1_line03.txt |
AC |
22 ms |
760 KB |
subtask1_line04.txt |
AC |
21 ms |
804 KB |
subtask1_line05.txt |
AC |
22 ms |
804 KB |
subtask1_line06.txt |
AC |
21 ms |
932 KB |
subtask1_random01.txt |
AC |
23 ms |
800 KB |
subtask1_random02.txt |
AC |
23 ms |
800 KB |
subtask1_random03.txt |
AC |
22 ms |
800 KB |
subtask1_random04.txt |
AC |
21 ms |
924 KB |
subtask1_random05.txt |
AC |
21 ms |
804 KB |
subtask1_random06.txt |
AC |
22 ms |
932 KB |
subtask1_random07.txt |
AC |
23 ms |
764 KB |
subtask1_random08.txt |
AC |
22 ms |
800 KB |
subtask1_special01.txt |
AC |
23 ms |
800 KB |
subtask1_special02.txt |
AC |
21 ms |
792 KB |
subtask1_special03.txt |
AC |
21 ms |
932 KB |
subtask1_special04.txt |
AC |
21 ms |
796 KB |