149552025-02-10 08:48:39sarminTúra (75 pont)cpp17Wrong answer 16/7554ms7988 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++) {
    int minus = 0;
    vector<int> mn;
    for (int& j : neighbors[i]) {
      if (mi[j] == -1) {
        minus++;
        mn.push_back(j);
      }
      if (mi[i] == -1 && mi[j] == -1) {
        cout << "NEM\n2\n" << i + 1 << " " << j + 1 << "\n"; return;
      }
      if (minus >= 2) {
        cout << "NEM\n3\n";
        cout << mn[0] << " " << i + 1 << " " << mn[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
base16/75
1Accepted0/01ms500 KiB
2Wrong answer0/01ms512 KiB
3Wrong answer0/61ms316 KiB
4Wrong answer0/34ms580 KiB
5Wrong answer0/34ms564 KiB
6Wrong answer0/44ms584 KiB
7Wrong answer0/44ms564 KiB
8Wrong answer0/44ms564 KiB
9Wrong answer0/241ms2156 KiB
10Accepted2/243ms2028 KiB
11Wrong answer0/241ms2148 KiB
12Wrong answer0/241ms2120 KiB
13Wrong answer0/241ms2020 KiB
14Wrong answer0/341ms2144 KiB
15Accepted2/241ms2108 KiB
16Accepted2/243ms2104 KiB
17Wrong answer0/241ms2020 KiB
18Wrong answer0/239ms2160 KiB
19Wrong answer0/339ms2120 KiB
20Wrong answer0/239ms2064 KiB
21Wrong answer0/239ms2036 KiB
22Wrong answer0/343ms2020 KiB
23Wrong answer0/250ms7988 KiB
24Accepted2/254ms7604 KiB
25Accepted2/248ms4564 KiB
26Wrong answer0/243ms4816 KiB
27Wrong answer0/248ms4556 KiB
28Wrong answer0/243ms4812 KiB
29Accepted2/248ms4600 KiB
30Wrong answer0/243ms4780 KiB
31Accepted2/248ms4832 KiB
32Accepted2/254ms4548 KiB