7892 | 2024. 01. 11 18:15:36 | Ablablabla | Túra (75 pont) | cpp17 | Elfogadva 75/75 | 64ms | 16348 KiB |
#include <bits/stdc++.h>
using namespace std;
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
int t;
cin >> t;
while(t--){
int n;
cin >> n;
vector<bool> jo(n, 0);
for(int i = 0; i < n; i++){
int a;
cin >> a;
jo[i] = (a == 1);
}
vector<vector<int>> csucsok(n, vector<int>());
for(int i = 0; i < n - 1; i++){
int a, b;
cin >> a >> b;
a--; b--;
csucsok[a].push_back(b);
csucsok[b].push_back(a);
}
bool rossz = 0;
for(int i = 0; i < n; i++){
if(jo[i]) continue;
for(int x : csucsok[i]){
if(!jo[x]){
cout << "NEM\n";
cout << "2\n";
cout << i + 1 << " " << x + 1 << "\n";
rossz = 1;
} else{
for(int y : csucsok[x]){
if(!jo[y] && y != i){
cout << "NEM\n";
cout << "3\n";
cout << i + 1 << " " << x + 1 << " " << y + 1 << "\n";
rossz = 1;
}
if(rossz){
break;
}
}
}
if(rossz){
break;
}
}
if(rossz){
break;
}
}
if(!rossz){
cout << "IGEN\n";
}
}
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 75/75 | ||||||
1 | Elfogadva | 0/0 | 3ms | 1828 KiB | |||
2 | Elfogadva | 0/0 | 3ms | 2064 KiB | |||
3 | Elfogadva | 6/6 | 3ms | 2232 KiB | |||
4 | Elfogadva | 3/3 | 6ms | 2508 KiB | |||
5 | Elfogadva | 3/3 | 6ms | 2732 KiB | |||
6 | Elfogadva | 4/4 | 6ms | 2688 KiB | |||
7 | Elfogadva | 4/4 | 7ms | 2944 KiB | |||
8 | Elfogadva | 4/4 | 7ms | 3152 KiB | |||
9 | Elfogadva | 2/2 | 43ms | 4388 KiB | |||
10 | Elfogadva | 2/2 | 45ms | 4448 KiB | |||
11 | Elfogadva | 2/2 | 43ms | 4704 KiB | |||
12 | Elfogadva | 2/2 | 43ms | 4656 KiB | |||
13 | Elfogadva | 2/2 | 43ms | 4976 KiB | |||
14 | Elfogadva | 3/3 | 43ms | 4968 KiB | |||
15 | Elfogadva | 2/2 | 45ms | 5020 KiB | |||
16 | Elfogadva | 2/2 | 43ms | 5080 KiB | |||
17 | Elfogadva | 2/2 | 43ms | 5292 KiB | |||
18 | Elfogadva | 2/2 | 41ms | 5484 KiB | |||
19 | Elfogadva | 3/3 | 41ms | 5436 KiB | |||
20 | Elfogadva | 2/2 | 41ms | 5440 KiB | |||
21 | Elfogadva | 2/2 | 43ms | 5432 KiB | |||
22 | Elfogadva | 3/3 | 46ms | 5400 KiB | |||
23 | Elfogadva | 2/2 | 48ms | 16348 KiB | |||
24 | Elfogadva | 2/2 | 64ms | 15296 KiB | |||
25 | Elfogadva | 2/2 | 48ms | 10076 KiB | |||
26 | Elfogadva | 2/2 | 46ms | 10704 KiB | |||
27 | Elfogadva | 2/2 | 50ms | 10476 KiB | |||
28 | Elfogadva | 2/2 | 46ms | 10796 KiB | |||
29 | Elfogadva | 2/2 | 50ms | 10288 KiB | |||
30 | Elfogadva | 2/2 | 46ms | 10796 KiB | |||
31 | Elfogadva | 2/2 | 52ms | 10624 KiB | |||
32 | Elfogadva | 2/2 | 54ms | 10284 KiB |