1866 | 2022. 12. 06 10:04:30 | kdb | Túra (75 pont) | cpp11 | Elfogadva 75/75 | 108ms | 16740 KiB |
#include <iostream>
#include <vector>
using namespace std;
int megoldas()
{
int m;
cin >> m;
vector<int> pontok(1, 0);
vector<int> rosszHelyek;
vector<vector<int>> elek(m + 1, vector<int>(0, 0));
for (int j = 0; j < m; j++)
{
int t;
cin >> t;
pontok.push_back(t);
if (t == -1) rosszHelyek.push_back(j + 1);
}
for (int j = 0; j < m - 1; j++)
{
int a, b;
cin >> a >> b;
elek[a].push_back(b);
elek[b].push_back(a);
}
for (int j : rosszHelyek)
{
for (int csucs : elek[j])
{
if (pontok[csucs] == -1)
{
cout << "NEM" << endl << 2 << endl << j << " " << csucs << endl;
return 0;
}
else
{
for (int harmadik : elek[csucs])
{
if (pontok[harmadik] == -1 && harmadik != j)
{
cout << "NEM" << endl << 3 << endl << j << " " << csucs << " " << harmadik << endl;
return 0;
}
}
}
}
}
cout << "IGEN" << endl;
return 0;
}
int main()
{
int n;
cin >> n;
for (size_t i = 0; i < n; i++)
{
megoldas();
}
return 0;
}
/*
2
4
-1 1 -1 -1
1 4
1 2
1 3
5
-1 1 1 -1 1
1 2
2 3
3 4
4 5
*/
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 75/75 | ||||||
1 | Elfogadva | 0/0 | 3ms | 1812 KiB | |||
2 | Elfogadva | 0/0 | 2ms | 2280 KiB | |||
3 | Elfogadva | 6/6 | 2ms | 2328 KiB | |||
4 | Elfogadva | 3/3 | 9ms | 2604 KiB | |||
5 | Elfogadva | 3/3 | 9ms | 2920 KiB | |||
6 | Elfogadva | 4/4 | 9ms | 3132 KiB | |||
7 | Elfogadva | 4/4 | 9ms | 3220 KiB | |||
8 | Elfogadva | 4/4 | 9ms | 3420 KiB | |||
9 | Elfogadva | 2/2 | 87ms | 4456 KiB | |||
10 | Elfogadva | 2/2 | 87ms | 4456 KiB | |||
11 | Elfogadva | 2/2 | 89ms | 4580 KiB | |||
12 | Elfogadva | 2/2 | 87ms | 4584 KiB | |||
13 | Elfogadva | 2/2 | 87ms | 4584 KiB | |||
14 | Elfogadva | 3/3 | 87ms | 4840 KiB | |||
15 | Elfogadva | 2/2 | 87ms | 5048 KiB | |||
16 | Elfogadva | 2/2 | 87ms | 5064 KiB | |||
17 | Elfogadva | 2/2 | 89ms | 4996 KiB | |||
18 | Elfogadva | 2/2 | 86ms | 5200 KiB | |||
19 | Elfogadva | 3/3 | 86ms | 5264 KiB | |||
20 | Elfogadva | 2/2 | 86ms | 5256 KiB | |||
21 | Elfogadva | 2/2 | 87ms | 5520 KiB | |||
22 | Elfogadva | 3/3 | 90ms | 5516 KiB | |||
23 | Elfogadva | 2/2 | 104ms | 16740 KiB | |||
24 | Elfogadva | 2/2 | 108ms | 16172 KiB | |||
25 | Elfogadva | 2/2 | 100ms | 10392 KiB | |||
26 | Elfogadva | 2/2 | 98ms | 10848 KiB | |||
27 | Elfogadva | 2/2 | 100ms | 10516 KiB | |||
28 | Elfogadva | 2/2 | 96ms | 11168 KiB | |||
29 | Elfogadva | 2/2 | 103ms | 10660 KiB | |||
30 | Elfogadva | 2/2 | 98ms | 11060 KiB | |||
31 | Elfogadva | 2/2 | 104ms | 10888 KiB | |||
32 | Elfogadva | 2/2 | 104ms | 10664 KiB |