149562025-02-10 08:56:14sarminTúra (75 pont)cpp17Accepted 75/7554ms6452 KiB
// Created by Armin on 2/10/2025.

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pp = pair<int, int>;
using vi = vector<int>;
using v = vector<int>;
using vvi = vector<vector<int>>;
using vpi = vector<pair<int, int>>;
using vl = vector<ll>;
constexpr ll MOD = 1e9 + 7;

void solve() {
  int n; cin >> n;
  vector<int> mi(n);
  for (int i = 0; i < n; i++) {
    cin >> mi[i];
  }
  vector<vector<int>> neighbors(n);
  for (int i = 0; i < n - 1; i++) {
    int a, b; cin >> a >> b;
    a--; b--;
    neighbors[a].push_back(b);
    neighbors[b].push_back(a);
  }

  for (int i = 0; i < n; i++) {
    vector<int> mn;
    for (int& j : neighbors[i]) {
      if (mi[j] == -1) {
        mn.push_back(j);
      }
      if (mi[i] == -1 && mi[j] == -1) {
        cout << "NEM\n2\n" << i + 1 << " " << j + 1 << "\n"; return;
      }
      if (mn.size() >= 2) {
        cout << "NEM\n3\n";
        cout << mn[0] + 1 << " " << i + 1 << " " << mn[1] + 1 << "\n"; return;
      }
    }
  }
  cout << "IGEN\n";
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(0);
    
    int t; cin >> t;
    while (t--) {
      solve();
    }
    
    return 0;
}
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/01ms508 KiB
2Accepted0/01ms316 KiB
3Accepted6/61ms316 KiB
4Accepted3/34ms316 KiB
5Accepted3/34ms316 KiB
6Accepted4/44ms316 KiB
7Accepted4/44ms316 KiB
8Accepted4/44ms480 KiB
9Accepted2/239ms824 KiB
10Accepted2/241ms1004 KiB
11Accepted2/241ms828 KiB
12Accepted2/241ms828 KiB
13Accepted2/241ms828 KiB
14Accepted3/341ms828 KiB
15Accepted2/241ms832 KiB
16Accepted2/241ms1016 KiB
17Accepted2/241ms1008 KiB
18Accepted2/239ms976 KiB
19Accepted3/339ms828 KiB
20Accepted2/237ms824 KiB
21Accepted2/241ms828 KiB
22Accepted3/343ms828 KiB
23Accepted2/248ms6452 KiB
24Accepted2/250ms6260 KiB
25Accepted2/248ms3396 KiB
26Accepted2/243ms3828 KiB
27Accepted2/248ms3388 KiB
28Accepted2/246ms3532 KiB
29Accepted2/248ms3272 KiB
30Accepted2/245ms3608 KiB
31Accepted2/248ms3536 KiB
32Accepted2/254ms3320 KiB