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